Spaces:

JosephZ
/

R1-SGG

Runtime error

JosephZ commited on Apr 27

Commit

1873cee

verified ·

1 Parent(s): b376a06

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -85,7 +85,7 @@ model = LLM(
             model=model_name,
             limit_mm_per_prompt={"image": 1},
             dtype='bfloat16',
-            device=device,
             max_model_len=4096,
             mm_processor_kwargs= { "max_pixels": 1024*28*28, "min_pixels": 4*28*28},
 )

             model=model_name,
             limit_mm_per_prompt={"image": 1},
             dtype='bfloat16',
+            #device=device,
             max_model_len=4096,
             mm_processor_kwargs= { "max_pixels": 1024*28*28, "min_pixels": 4*28*28},
 )