EthanReid commited on
Commit
4489767
·
1 Parent(s): 362bf44

gpu in readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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
- # device_map={"": "cuda"}
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