deepseekmirror/DeepSeek-Coder
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-Coder.git synced 2025-04-19 18:19:05 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
ZHU QIHAO 2023-10-26 18:43:01 +08:00 committed by GitHub
parent 83eb2d811b
commit 8ba4cdcb63
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -144,6 +144,8 @@ inputs = tokenizer.encode(input_text, return_tensors="pt").to(device)
outputs = model.generate(**inputs, max_length=128) outputs = model.generate(**inputs, max_length=128)
print(tokenizer.decode(outputs[0])) print(tokenizer.decode(outputs[0]))
``` ```
![Completion GIF](pictures/completion_demo.gif)
#### Chat Model Inference #### Chat Model Inference
```python ```python
from transformers import AutoTokenizer, AutoModelForCausalLM from transformers import AutoTokenizer, AutoModelForCausalLM