Spaces:
Running
on
Zero
Running
on
Zero
John Ho
commited on
Commit
·
b89bc96
1
Parent(s):
d9bd1e8
adding more internvl3 models
Browse files
app.py
CHANGED
|
@@ -117,6 +117,12 @@ MODEL_ZOO = {
|
|
| 117 |
"InternVL3-1B-hf": AutoModelForImageTextToText.from_pretrained(
|
| 118 |
"OpenGVLab/InternVL3-1B-hf", device_map=DEVICE, torch_dtype=DTYPE
|
| 119 |
),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 120 |
}
|
| 121 |
|
| 122 |
PROCESSORS = {
|
|
@@ -124,6 +130,8 @@ PROCESSORS = {
|
|
| 124 |
"qwen2.5-vl-7b-instruct": load_processor("Qwen/Qwen2.5-VL-7B-Instruct"),
|
| 125 |
"qwen2.5-vl-3b-instruct": load_processor("Qwen/Qwen2.5-VL-3B-Instruct"),
|
| 126 |
"InternVL3-1B-hf": load_processor("OpenGVLab/InternVL3-1B-hf"),
|
|
|
|
|
|
|
| 127 |
}
|
| 128 |
logger.debug("Models and Processors Loaded!")
|
| 129 |
|
|
|
|
| 117 |
"InternVL3-1B-hf": AutoModelForImageTextToText.from_pretrained(
|
| 118 |
"OpenGVLab/InternVL3-1B-hf", device_map=DEVICE, torch_dtype=DTYPE
|
| 119 |
),
|
| 120 |
+
"InternVL3-2B-hf": AutoModelForImageTextToText.from_pretrained(
|
| 121 |
+
"OpenGVLab/InternVL3-2B-hf", device_map=DEVICE, torch_dtype=DTYPE
|
| 122 |
+
),
|
| 123 |
+
"InternVL3-8B-hf": AutoModelForImageTextToText.from_pretrained(
|
| 124 |
+
"OpenGVLab/InternVL3-8B-hf", device_map=DEVICE, torch_dtype=DTYPE
|
| 125 |
+
),
|
| 126 |
}
|
| 127 |
|
| 128 |
PROCESSORS = {
|
|
|
|
| 130 |
"qwen2.5-vl-7b-instruct": load_processor("Qwen/Qwen2.5-VL-7B-Instruct"),
|
| 131 |
"qwen2.5-vl-3b-instruct": load_processor("Qwen/Qwen2.5-VL-3B-Instruct"),
|
| 132 |
"InternVL3-1B-hf": load_processor("OpenGVLab/InternVL3-1B-hf"),
|
| 133 |
+
"InternVL3-2B-hf": load_processor("OpenGVLab/InternVL3-2B-hf"),
|
| 134 |
+
"InternVL3-8B-hf": load_processor("OpenGVLab/InternVL3-8B-hf"),
|
| 135 |
}
|
| 136 |
logger.debug("Models and Processors Loaded!")
|
| 137 |
|