Update README.md
Browse files
README.md
CHANGED
@@ -125,7 +125,7 @@ import torch
|
|
125 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
126 |
|
127 |
model = AutoModelForCausalLM.from_pretrained("DiscoResearch/DiscoLM-mixtral-8x7b-v2", low_cpu_mem_usage=True, device_map="auto", trust_remote_code=True)
|
128 |
-
|
129 |
chat = [
|
130 |
{"role": "system", "content": "You are DiscoLM, a helpful assistant."},
|
131 |
{"role": "user", "content": "Please tell me possible reasons to call a research collective Disco Research"}
|
|
|
125 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
126 |
|
127 |
model = AutoModelForCausalLM.from_pretrained("DiscoResearch/DiscoLM-mixtral-8x7b-v2", low_cpu_mem_usage=True, device_map="auto", trust_remote_code=True)
|
128 |
+
tokenizer = AutoTokenizer.from_pretrained("DiscoResearch/DiscoLM-mixtral-8x7b-v2")
|
129 |
chat = [
|
130 |
{"role": "system", "content": "You are DiscoLM, a helpful assistant."},
|
131 |
{"role": "user", "content": "Please tell me possible reasons to call a research collective Disco Research"}
|