Spaces:
Runtime error
Runtime error
add 7b model
Browse files
app.py
CHANGED
|
@@ -33,15 +33,15 @@ def array_to_image_path(image_array):
|
|
| 33 |
return full_path
|
| 34 |
|
| 35 |
models = {
|
| 36 |
-
"qwen2-vl-fmb": Qwen2VLForConditionalGeneration.from_pretrained("fmb-quibdo/qwen2-vl-fmb", torch_dtype="auto").cuda().eval()
|
| 37 |
-
|
| 38 |
}
|
| 39 |
|
| 40 |
processors = {
|
| 41 |
"qwen2-vl-fmb": AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
|
| 42 |
}
|
| 43 |
|
| 44 |
-
DESCRIPTION = "[Qwen2-VL
|
| 45 |
|
| 46 |
kwargs = {}
|
| 47 |
kwargs['torch_dtype'] = torch.bfloat16
|
|
|
|
| 33 |
return full_path
|
| 34 |
|
| 35 |
models = {
|
| 36 |
+
"qwen2-vl-fmb": Qwen2VLForConditionalGeneration.from_pretrained("fmb-quibdo/qwen2-vl-fmb", torch_dtype="auto").cuda().eval(),
|
| 37 |
+
"qwen2-vl-fmb-7B": Qwen2VLForConditionalGeneration.from_pretrained("fmb-quibdo/fmb-qwen-vl-7b", torch_dtype="auto").cuda().eval(),
|
| 38 |
}
|
| 39 |
|
| 40 |
processors = {
|
| 41 |
"qwen2-vl-fmb": AutoProcessor.from_pretrained("Qwen/Qwen2-VL-7B-Instruct")
|
| 42 |
}
|
| 43 |
|
| 44 |
+
DESCRIPTION = "[Qwen2-VL 2B or 7B Spanish Demo](https://huggingface.co/ajanco/qwen2-vl-fmb)"
|
| 45 |
|
| 46 |
kwargs = {}
|
| 47 |
kwargs['torch_dtype'] = torch.bfloat16
|