prithivMLmods commited on
Commit
4fe3125
·
verified ·
1 Parent(s): 0ff76c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -59,10 +59,12 @@ model_o = Qwen2VLForConditionalGeneration.from_pretrained(
59
 
60
  # Load Lumian-VLR-7B-Thinking
61
  MODEL_ID_J = "prithivMLmods/Lumian-VLR-7B-Thinking"
62
- processor_j = AutoProcessor.from_pretrained(MODEL_ID_J, trust_remote_code=True)
 
63
  model_j = Qwen2_5_VLForConditionalGeneration.from_pretrained(
64
  MODEL_ID_J,
65
  trust_remote_code=True,
 
66
  torch_dtype=torch.float16
67
  ).to(device).eval()
68
 
 
59
 
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