gaze-demo

Runtime error

soiz1 commited on Jan 18

Commit

895cfa4

verified ·

1 Parent(s): 2037538

Update demo.py

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -15,7 +15,7 @@ model = AutoModelForCausalLM.from_pretrained(
     "vikhyatk/moondream-next",
     trust_remote_code=True,
     torch_dtype=torch.float16,
-    device_map={"": "cuda"},
     revision="64ce9d45e8603058826ab339d3bb0f48f7289b1f"
 )

     "vikhyatk/moondream-next",
     trust_remote_code=True,
     torch_dtype=torch.float16,
+    device_map={"": "cpu"},
     revision="64ce9d45e8603058826ab339d3bb0f48f7289b1f"
 )