Update README.md
Browse files
README.md
CHANGED
|
@@ -33,7 +33,7 @@ from transformers import AutoModel
|
|
| 33 |
import torch
|
| 34 |
|
| 35 |
model = AutoModel.from_pretrained(
|
| 36 |
-
"
|
| 37 |
trust_remote_code=True,
|
| 38 |
).to("cuda" if torch.cuda.is_available() else "cpu")
|
| 39 |
|
|
|
|
| 33 |
import torch
|
| 34 |
|
| 35 |
model = AutoModel.from_pretrained(
|
| 36 |
+
"QuarkML/QaDiT", # or a local export folder
|
| 37 |
trust_remote_code=True,
|
| 38 |
).to("cuda" if torch.cuda.is_available() else "cpu")
|
| 39 |
|