Chickaboo commited on
Commit
fc66abc
·
verified ·
1 Parent(s): c4fa1e9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -3
README.md CHANGED
@@ -14,9 +14,10 @@ math expert,- **meta-math/MetaMath-Mistral-7B**
14
 
15
  ### Usage
16
 
17
- ```from transformers import AutoModelForCausalLM, AutoTokenizer
 
18
 
19
- model_id = "mistral-community/Mixtral-8x22B-v0.1"
20
  tokenizer = AutoTokenizer.from_pretrained(model_id)
21
 
22
  model = AutoModelForCausalLM.from_pretrained(model_id)
@@ -25,4 +26,5 @@ text = "Hello my name is"
25
  inputs = tokenizer(text, return_tensors="pt")
26
 
27
  outputs = model.generate(**inputs, max_new_tokens=20)
28
- print(tokenizer.decode(outputs[0], skip_special_tokens=True))```
 
 
14
 
15
  ### Usage
16
 
17
+ ```python
18
+ from transformers import AutoModelForCausalLM, AutoTokenizer
19
 
20
+ model_id = "Chickaboo/Chicka-Mistral-4x7b"
21
  tokenizer = AutoTokenizer.from_pretrained(model_id)
22
 
23
  model = AutoModelForCausalLM.from_pretrained(model_id)
 
26
  inputs = tokenizer(text, return_tensors="pt")
27
 
28
  outputs = model.generate(**inputs, max_new_tokens=20)
29
+ print(tokenizer.decode(outputs[0], skip_special_tokens=True))
30
+ ```