Spaces:
Sleeping
Sleeping
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
|