Update README.md
Browse files
README.md
CHANGED
@@ -39,7 +39,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
39 |
torch.set_default_device("cuda")
|
40 |
|
41 |
model = AutoModelForCausalLM.from_pretrained("ibivibiv/strix-rufipes-70b", torch_dtype="auto", device_config='auto')
|
42 |
-
tokenizer = AutoTokenizer.from_pretrained("ibivibiv/strix-rufipes-70b"
|
43 |
|
44 |
inputs = tokenizer("### Instruction: Create a plan for developing the game of snake in python using pygame.\n### Response:\n", return_tensors="pt", return_attention_mask=False)
|
45 |
|
|
|
39 |
torch.set_default_device("cuda")
|
40 |
|
41 |
model = AutoModelForCausalLM.from_pretrained("ibivibiv/strix-rufipes-70b", torch_dtype="auto", device_config='auto')
|
42 |
+
tokenizer = AutoTokenizer.from_pretrained("ibivibiv/strix-rufipes-70b")
|
43 |
|
44 |
inputs = tokenizer("### Instruction: Create a plan for developing the game of snake in python using pygame.\n### Response:\n", return_tensors="pt", return_attention_mask=False)
|
45 |
|