jiang-cc commited on
Commit
86c4571
·
1 Parent(s): 4c8234a

fix: use AutoModelForImageTextToText, add spaces dep

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +1 -0
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  import spaces
7
  import gradio as gr
8
  import torch
9
- from transformers import AutoModelForVision2Seq, AutoProcessor
10
  from qwen_vl_utils import process_vision_info
11
  from PIL import Image
12
 
@@ -23,7 +23,7 @@ processor = AutoProcessor.from_pretrained(
23
  trust_remote_code=True,
24
  )
25
 
26
- model = AutoModelForVision2Seq.from_pretrained(
27
  MODEL_ID,
28
  torch_dtype=torch.bfloat16,
29
  device_map="auto",
 
6
  import spaces
7
  import gradio as gr
8
  import torch
9
+ from transformers import AutoModelForImageTextToText, AutoProcessor
10
  from qwen_vl_utils import process_vision_info
11
  from PIL import Image
12
 
 
23
  trust_remote_code=True,
24
  )
25
 
26
+ model = AutoModelForImageTextToText.from_pretrained(
27
  MODEL_ID,
28
  torch_dtype=torch.bfloat16,
29
  device_map="auto",
requirements.txt CHANGED
@@ -3,3 +3,4 @@ transformers>=4.55.0
3
  accelerate
4
  qwen_vl_utils
5
  Pillow
 
 
3
  accelerate
4
  qwen_vl_utils
5
  Pillow
6
+ spaces