Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ logging.basicConfig(level=logging.INFO)
|
|
18 |
logger = logging.getLogger(__name__)
|
19 |
|
20 |
# Load model and processor
|
21 |
-
ckpt ="Qwen/Qwen2-VL-
|
22 |
model = Qwen2VLForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16,trust_remote_code=True).to("cuda")
|
23 |
processor = AutoProcessor.from_pretrained(ckpt,trust_remote_code=True)
|
24 |
|
|
|
18 |
logger = logging.getLogger(__name__)
|
19 |
|
20 |
# Load model and processor
|
21 |
+
ckpt ="Qwen/Qwen2-VL-7B-Instruct-AWQ"
|
22 |
model = Qwen2VLForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16,trust_remote_code=True).to("cuda")
|
23 |
processor = AutoProcessor.from_pretrained(ckpt,trust_remote_code=True)
|
24 |
|