ForeverBlue commited on
Commit
640a51d
路
verified 路
1 Parent(s): 040aef8

Add Qwen3-VL vision runtime dependency

Browse files
Files changed (2) hide show
  1. app.py +2 -3
  2. requirements.txt +1 -0
app.py CHANGED
@@ -3,13 +3,13 @@ import threading
3
  import gradio as gr
4
  import spaces
5
  import torch
6
- from transformers import AutoModelForMultimodalLM, AutoProcessor
7
 
8
 
9
  MODEL_ID = "ForeverBlue/Qwen3-VL-2B-GRACE-BF16"
10
 
11
  processor = AutoProcessor.from_pretrained(MODEL_ID)
12
- model = AutoModelForMultimodalLM.from_pretrained(
13
  MODEL_ID,
14
  dtype=torch.bfloat16,
15
  low_cpu_mem_usage=True,
@@ -99,4 +99,3 @@ with gr.Blocks(title="GRACE-VLM Demo") as demo:
99
 
100
  if __name__ == "__main__":
101
  demo.queue(default_concurrency_limit=1).launch()
102
-
 
3
  import gradio as gr
4
  import spaces
5
  import torch
6
+ from transformers import AutoProcessor, Qwen3VLForConditionalGeneration
7
 
8
 
9
  MODEL_ID = "ForeverBlue/Qwen3-VL-2B-GRACE-BF16"
10
 
11
  processor = AutoProcessor.from_pretrained(MODEL_ID)
12
+ model = Qwen3VLForConditionalGeneration.from_pretrained(
13
  MODEL_ID,
14
  dtype=torch.bfloat16,
15
  low_cpu_mem_usage=True,
 
99
 
100
  if __name__ == "__main__":
101
  demo.queue(default_concurrency_limit=1).launch()
 
requirements.txt CHANGED
@@ -1,4 +1,5 @@
1
  accelerate==1.13.0
2
  pillow>=11,<12
3
  torch==2.8.0
 
4
  transformers==5.9.0
 
1
  accelerate==1.13.0
2
  pillow>=11,<12
3
  torch==2.8.0
4
+ torchvision==0.23.0
5
  transformers==5.9.0