Spaces:
Running on Zero
Running on Zero
fix: use AutoModelForImageTextToText, add spaces dep
Browse files- app.py +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
|
| 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 =
|
| 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
|