Spaces:

GF-John
/

video-caption

Running on Zero

John Ho commited on Jul 24

Commit

c7e712e

1 Parent(s): c697b34

fixing issue with device map for the inputs

Files changed (1) hide show

app.py CHANGED Viewed

@@ -76,7 +76,7 @@ def load_model(
 def load_processor(model_name="Qwen/Qwen2.5-VL-7B-Instruct"):
     return AutoProcessor.from_pretrained(
         model_name,
-        device_map=DEVICE,
         use_fast=True,
         torch_dtype=DTYPE,
     )

 def load_processor(model_name="Qwen/Qwen2.5-VL-7B-Instruct"):
     return AutoProcessor.from_pretrained(
         model_name,
+        # device_map=DEVICE,
         use_fast=True,
         torch_dtype=DTYPE,
     )