John Ho commited on
Commit
b89bc96
·
1 Parent(s): d9bd1e8

adding more internvl3 models

Browse files
Files changed (1) hide show
  1. app.py +8 -0
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