Update README.md
Browse files
README.md
CHANGED
|
@@ -56,7 +56,7 @@ processor = AutoProcessor.from_pretrained(
|
|
| 56 |
revision="827103c", # optional
|
| 57 |
)
|
| 58 |
model = AutoModelForPreTraining.from_pretrained(
|
| 59 |
-
|
| 60 |
trust_remote_code=True,
|
| 61 |
revision="827103c", # optional
|
| 62 |
torch_dtype=torch.bfloat16
|
|
|
|
| 56 |
revision="827103c", # optional
|
| 57 |
)
|
| 58 |
model = AutoModelForPreTraining.from_pretrained(
|
| 59 |
+
REPO,
|
| 60 |
trust_remote_code=True,
|
| 61 |
revision="827103c", # optional
|
| 62 |
torch_dtype=torch.bfloat16
|