EthanReid
commited on
Commit
·
4489767
1
Parent(s):
362bf44
gpu in readme
Browse files
README.md
CHANGED
@@ -24,7 +24,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
24 |
"moondream/moondream-2b-2025-04-14-4bit",
|
25 |
trust_remote_code=True,
|
26 |
# Uncomment to run on GPU.
|
27 |
-
|
28 |
)
|
29 |
|
30 |
# Captioning
|
|
|
24 |
"moondream/moondream-2b-2025-04-14-4bit",
|
25 |
trust_remote_code=True,
|
26 |
# Uncomment to run on GPU.
|
27 |
+
device_map={"": "cuda"}
|
28 |
)
|
29 |
|
30 |
# Captioning
|