Chickaboo commited on
Commit
87c519c
·
verified ·
1 Parent(s): e3ab2fc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -81,8 +81,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
81
 
82
  device = "cuda" # the device to load the model onto
83
 
84
- model = AutoModelForCausalLM.from_pretrained("Chickaboo/Chicka-Mistral-4x7b")
85
- tokenizer = AutoTokenizer.from_pretrained("Chickaboo/Chicka-Mistral-4x7b")
86
 
87
  messages = [
88
  {"role": "user", "content": "What is your favourite condiment?"},
 
81
 
82
  device = "cuda" # the device to load the model onto
83
 
84
+ model = AutoModelForCausalLM.from_pretrained("Chickaboo/Chicka-Mistral-3x7b")
85
+ tokenizer = AutoTokenizer.from_pretrained("Chickaboo/Chicka-Mixtral-3x7b")
86
 
87
  messages = [
88
  {"role": "user", "content": "What is your favourite condiment?"},