Hugging Face's logo

Spaces:
omlab
/
VLM-R1-Referral-Expression
Running on T4

App Files Community
2
SZhanZ commited on
Commit
3121838
·
1 Parent(s): 4b25987

fp16

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -76,7 +76,7 @@ if __name__ == "__main__":
76
 
77
  # model_path = "/data/shz/project/vlm-r1/VLM-R1/output/Qwen2.5-VL-3B-GRPO-REC/checkpoint-500"
78
  model_path = "SZhanZ/Qwen2.5VL-VLM-R1-REC-step500"
79
- model = Qwen2_5_VLForConditionalGeneration.from_pretrained(model_path)
80
  processor = AutoProcessor.from_pretrained(model_path)
81
 
82
  def gradio_interface(image, text):
 
76
 
77
  # model_path = "/data/shz/project/vlm-r1/VLM-R1/output/Qwen2.5-VL-3B-GRPO-REC/checkpoint-500"
78
  model_path = "SZhanZ/Qwen2.5VL-VLM-R1-REC-step500"
79
+ model = Qwen2_5_VLForConditionalGeneration.from_pretrained(model_path, torch_dtype=torch.float16)
80
  processor = AutoProcessor.from_pretrained(model_path)
81
 
82
  def gradio_interface(image, text):