aiqtech commited on
Commit
9256bf2
·
verified ·
1 Parent(s): c462fef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import subprocess
11
  subprocess.run('pip install --upgrade transformers', shell=True)
12
  subprocess.run('pip install accelerate', shell=True)
13
 
14
- from transformers import AutoProcessor, AutoModelForVisionText2Text
15
 
16
  # Model and processor initialization with trust_remote_code=True
17
  processor = AutoProcessor.from_pretrained(
@@ -19,7 +19,7 @@ processor = AutoProcessor.from_pretrained(
19
  trust_remote_code=True
20
  )
21
 
22
- model = AutoModelForVisionText2Text.from_pretrained(
23
  "Qwen/QVQ-72B-Preview",
24
  trust_remote_code=True,
25
  device_map="auto"
 
11
  subprocess.run('pip install --upgrade transformers', shell=True)
12
  subprocess.run('pip install accelerate', shell=True)
13
 
14
+ from transformers import AutoProcessor, AutoModelForVisionEncoderDecoder
15
 
16
  # Model and processor initialization with trust_remote_code=True
17
  processor = AutoProcessor.from_pretrained(
 
19
  trust_remote_code=True
20
  )
21
 
22
+ model = AutoModelForVisionEncoderDecoder.from_pretrained(
23
  "Qwen/QVQ-72B-Preview",
24
  trust_remote_code=True,
25
  device_map="auto"