Update README.md

#3
by Remek - opened
Files changed (1) hide show
  1. README.md +1 -1
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
- tok = 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"}
 
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"}