Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -281,7 +281,7 @@ Note: To use function calling, you should see the github repo above.
281
 
282
  import torch
283
  from transformers import AutoTokenizer, AutoModelForCausalLM
284
- from transformers import LlamaTokenizer, MixtralForCausalLM
285
  import bitsandbytes, flash_attn
286
 
287
  tokenizer = LlamaTokenizer.from_pretrained('NousResearch/Hermes-2-Pro-Mistral-7B', trust_remote_code=True)
 
281
 
282
  import torch
283
  from transformers import AutoTokenizer, AutoModelForCausalLM
284
+ from transformers import LlamaTokenizer, MistralForCausalLM
285
  import bitsandbytes, flash_attn
286
 
287
  tokenizer = LlamaTokenizer.from_pretrained('NousResearch/Hermes-2-Pro-Mistral-7B', trust_remote_code=True)