TimeRobber dataviral commited on
Commit
a8d85f7
·
1 Parent(s): 235b80c

Readme from_pretrained model version fix (#18)

Browse files

- Readme from_pretrained model version fix (4208713cc4a830ecc383f5584e969b4f1fb020af)


Co-authored-by: Aviral Joshi <[email protected]>

Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -39,8 +39,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
39
 
40
  device = "cuda" # the device to load the model onto
41
 
42
- model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
43
- tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.1")
44
 
45
  messages = [
46
  {"role": "user", "content": "What is your favourite condiment?"},
 
39
 
40
  device = "cuda" # the device to load the model onto
41
 
42
+ model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
43
+ tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-Instruct-v0.2")
44
 
45
  messages = [
46
  {"role": "user", "content": "What is your favourite condiment?"},