Spaces:

lightonai
/

vision_pipeline

Running on Zero

Antoine Chaffin commited on Sep 27

Commit

8560547

•

1 Parent(s): 5885f58

Back to 2B

Files changed (1) hide show

app.py CHANGED Viewed

@@ -20,14 +20,14 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # Initialize the model and processor
 model = (
     Qwen2VLForConditionalGeneration.from_pretrained(
-        "Qwen/Qwen2-VL-7B-Instruct", trust_remote_code=True, torch_dtype=torch.bfloat16
     )
     .to(device)
     .eval()
 )
 processor = AutoProcessor.from_pretrained(
-    "Qwen/Qwen2-VL-7B-Instruct", trust_remote_code=True
 )

 # Initialize the model and processor
 model = (
     Qwen2VLForConditionalGeneration.from_pretrained(
+        "Qwen/Qwen2-VL-2B-Instruct", trust_remote_code=True, torch_dtype=torch.bfloat16
     )
     .to(device)
     .eval()
 )
 processor = AutoProcessor.from_pretrained(
+    "Qwen/Qwen2-VL-2B-Instruct", trust_remote_code=True
 )