Commit
•
69bec6a
1
Parent(s):
59685c3
Fix typo in usage
Browse files
README.md
CHANGED
|
@@ -25,7 +25,7 @@ from sentencepiece import SentencePieceProcessor
|
|
| 25 |
|
| 26 |
torch.set_default_dtype(torch.bfloat16)
|
| 27 |
model = AutoModelForCausalLM.from_pretrained(
|
| 28 |
-
"
|
| 29 |
trust_remote_code=True,
|
| 30 |
device_map="auto",
|
| 31 |
torch_dtype=torch.bfloat16,
|
|
|
|
| 25 |
|
| 26 |
torch.set_default_dtype(torch.bfloat16)
|
| 27 |
model = AutoModelForCausalLM.from_pretrained(
|
| 28 |
+
"hpcai-tech/grok-1",
|
| 29 |
trust_remote_code=True,
|
| 30 |
device_map="auto",
|
| 31 |
torch_dtype=torch.bfloat16,
|