cloudyu commited on
Commit
4358b93
1 Parent(s): b58a6fd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
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
+ ```