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 |
+
```
|