OpenBuddy/openbuddy-atom-13b-v9-bf16
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Cherrytest
2023-08-22 12:50:17 +00:00
parent d0aaf9ac57
commit 5f187f631f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -46,7 +46,7 @@ You are trained by OpenBuddy team, (https://openbuddy.ai, https://github.com/Ope
User: 睡不着怎么办?
Assistant:"""
inputs = tokenizer.encode(prompt, return_tensors="pt")
inputs = tokenizer.encode(prompt, return_tensors="pt").cuda()
outputs = model.generate(inputs, max_length=512)
response = tokenizer.decode(outputs[0])
print(response)