Update README.md
Browse files
README.md
CHANGED
|
@@ -14,6 +14,7 @@ license_link: https://huggingface.co/01-ai/Yi-34B-200K/blob/main/LICENSE
|
|
| 14 |
* Metrics not Test
|
| 15 |
|
| 16 |
code example
|
|
|
|
| 17 |
```
|
| 18 |
import torch
|
| 19 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
@@ -34,4 +35,5 @@ while len(prompt) > 0:
|
|
| 34 |
input_ids=input_ids, max_new_tokens=1500,repetition_penalty=1.1
|
| 35 |
)
|
| 36 |
print(tokenizer.decode(generation_output[0]))
|
| 37 |
-
prompt = input("please input prompt:")
|
|
|
|
|
|
| 14 |
* Metrics not Test
|
| 15 |
|
| 16 |
code example
|
| 17 |
+
|
| 18 |
```
|
| 19 |
import torch
|
| 20 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
|
| 35 |
input_ids=input_ids, max_new_tokens=1500,repetition_penalty=1.1
|
| 36 |
)
|
| 37 |
print(tokenizer.decode(generation_output[0]))
|
| 38 |
+
prompt = input("please input prompt:")
|
| 39 |
+
```
|