aloobun commited on
Commit
9c269e6
1 Parent(s): 23a60c8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -1
README.md CHANGED
@@ -13,6 +13,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig
13
 
14
  tokenizer = AutoTokenizer.from_pretrained("qwen-1_8b-bun-chat", trust_remote_code=True)
15
  model = AutoModelForCausalLM.from_pretrained("qwen-1_8b-bun-chat", device_map="auto", trust_remote_code=True).eval()
16
- response, history = model.chat(tokenizer, "Hi, how are you?", history=history)
17
  print(response)
 
 
 
18
  ```
 
13
 
14
  tokenizer = AutoTokenizer.from_pretrained("qwen-1_8b-bun-chat", trust_remote_code=True)
15
  model = AutoModelForCausalLM.from_pretrained("qwen-1_8b-bun-chat", device_map="auto", trust_remote_code=True).eval()
16
+ response, _ = model.chat(tokenizer, "Hi, how are you?", history=None)
17
  print(response)
18
+
19
+ #response, history = model.chat(tokenizer, "I'm so happy with my new wardrobe! The clothes I bought fit me perfectly and they look great.", history=history)
20
+ #print(response)
21
  ```