Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|
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 =
|
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"
|