Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -60,11 +60,11 @@ model_o = Qwen2VLForConditionalGeneration.from_pretrained(
|
|
60 |
# Load Lumian-VLR-7B-Thinking
|
61 |
MODEL_ID_J = "prithivMLmods/Lumian-VLR-7B-Thinking"
|
62 |
SUBFOLDER = "think-preview"
|
63 |
-
processor_j = AutoProcessor.from_pretrained(MODEL_ID_J, trust_remote_code=True, SUBFOLDER)
|
64 |
model_j = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
65 |
MODEL_ID_J,
|
66 |
trust_remote_code=True,
|
67 |
-
SUBFOLDER,
|
68 |
torch_dtype=torch.float16
|
69 |
).to(device).eval()
|
70 |
|
|
|
60 |
# Load Lumian-VLR-7B-Thinking
|
61 |
MODEL_ID_J = "prithivMLmods/Lumian-VLR-7B-Thinking"
|
62 |
SUBFOLDER = "think-preview"
|
63 |
+
processor_j = AutoProcessor.from_pretrained(MODEL_ID_J, trust_remote_code=True, subfolder=SUBFOLDER)
|
64 |
model_j = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
65 |
MODEL_ID_J,
|
66 |
trust_remote_code=True,
|
67 |
+
subfolder=SUBFOLDER,
|
68 |
torch_dtype=torch.float16
|
69 |
).to(device).eval()
|
70 |
|