Update README.md
Browse files
README.md
CHANGED
|
@@ -20,8 +20,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
| 20 |
"vikhyatk/moondream2",
|
| 21 |
revision="2025-06-21",
|
| 22 |
trust_remote_code=True,
|
| 23 |
-
#
|
| 24 |
-
# device_map={"": "cuda"}
|
| 25 |
)
|
| 26 |
|
| 27 |
# Captioning
|
|
|
|
| 20 |
"vikhyatk/moondream2",
|
| 21 |
revision="2025-06-21",
|
| 22 |
trust_remote_code=True,
|
| 23 |
+
device_map={"": "cuda"} # ...or 'mps', on Apple Silicon
|
|
|
|
| 24 |
)
|
| 25 |
|
| 26 |
# Captioning
|