Spaces:

fumucu
/

LAOD

Running on Zero

fumucu commited on Aug 9

Commit

2a2872c

1 Parent(s): 6534772

initial commit

Files changed (1) hide show

laod_pipeline.py CHANGED Viewed

@@ -21,7 +21,7 @@ pipe = pipeline(
     "image-text-to-text",
     model="google/gemma-3-4b-it",
     #device="cuda:1",
-    device_map=device',
     torch_dtype=torch.bfloat16
 )

     "image-text-to-text",
     model="google/gemma-3-4b-it",
     #device="cuda:1",
+    device_map=device,
     torch_dtype=torch.bfloat16
 )