Update README.md
Browse files
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,
|
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 |
```
|