Update README.md
Browse files
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-
|
52 |
-
tokenizer = AutoTokenizer.from_pretrained("ibivibiv/aegolius-acadicus-
|
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 |
|