ibivibiv commited on
Commit
040bcca
1 Parent(s): b3b9ce4

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -48,8 +48,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
48
 
49
  torch.set_default_device("cuda")
50
 
51
- model = AutoModelForCausalLM.from_pretrained("ibivibiv/aegolius-acadicus-24b-v2", torch_dtype="auto", device_config='auto')
52
- tokenizer = AutoTokenizer.from_pretrained("ibivibiv/aegolius-acadicus-24b-v2")
53
 
54
  inputs = tokenizer("### Instruction: Who would when in an arm wrestling match between Abraham Lincoln and Chuck Norris?\n### Response:\n", return_tensors="pt", return_attention_mask=False)
55
 
 
48
 
49
  torch.set_default_device("cuda")
50
 
51
+ model = AutoModelForCausalLM.from_pretrained("ibivibiv/aegolius-acadicus-34b-v3", torch_dtype="auto", device_config='auto')
52
+ tokenizer = AutoTokenizer.from_pretrained("ibivibiv/aegolius-acadicus-34b-v3")
53
 
54
  inputs = tokenizer("### Instruction: Who would when in an arm wrestling match between Abraham Lincoln and Chuck Norris?\n### Response:\n", return_tensors="pt", return_attention_mask=False)
55