This commit is contained in:
P小二
2024-03-22 00:30:07 +08:00
parent 47c0e23250
commit ce9f6b6145
3 changed files with 26 additions and 4 deletions

5
.idea/.gitignore generated vendored Normal file
View File

@@ -0,0 +1,5 @@
# 默认忽略的文件
/shelf/
/workspace.xml
# GitHub Copilot persisted chat sessions
/copilot/chatSessions

View File

@@ -1,11 +1,10 @@
# pip install accelerate
from transformers import AutoTokenizer, AutoModelForCausalLM from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b-it") tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b-it")
model = AutoModelForCausalLM.from_pretrained("google/gemma-7b-it", device_map="auto") model = AutoModelForCausalLM.from_pretrained("google/gemma-7b-it")
input_text = "Write me a poem about Machine Learning." input_text = "Write me a poem about Machine Learning."
input_ids = tokenizer(input_text, return_tensors="pt").to("cuda") input_ids = tokenizer(input_text, return_tensors="pt")
outputs = model.generate(**input_ids) outputs = model.generate(**input_ids)
print(tokenizer.decode(outputs[0])) print(tokenizer.decode(outputs[0]))

View File

@@ -1,2 +1,20 @@
# FlyPythonLLM # FlyPythonLLM
Code of FlyPython Newsletter
Code of [FlyPython Newsletter](https://blog.flypython.com/)
## free content
1. GPTChat
2. ChatWithGemma7b
3. GPTs Action (1)
4. GPTs Action (2)
5. LangChain(1)
6. LangChain(2)
## paid content
1. MoneyPrinter
2. mindgraph
3. metavoice