Spaces:
Running
on
Zero
Running
on
Zero
John Ho
commited on
Commit
·
5a0f6f1
1
Parent(s):
3feaae8
try to load gemma again
Browse files
app.py
CHANGED
@@ -146,11 +146,11 @@ MODEL_ZOO = {
|
|
146 |
use_flash_attention=False,
|
147 |
apply_quantization=True,
|
148 |
),
|
149 |
-
|
150 |
-
|
151 |
-
|
152 |
-
|
153 |
-
|
154 |
}
|
155 |
|
156 |
PROCESSORS = {
|
@@ -160,7 +160,7 @@ PROCESSORS = {
|
|
160 |
"InternVL3-1B-hf": load_processor("OpenGVLab/InternVL3-1B-hf"),
|
161 |
"InternVL3-2B-hf": load_processor("OpenGVLab/InternVL3-2B-hf"),
|
162 |
"InternVL3-8B-hf": load_processor("OpenGVLab/InternVL3-8B-hf"),
|
163 |
-
|
164 |
}
|
165 |
logger.debug("Models and Processors Loaded!")
|
166 |
|
|
|
146 |
use_flash_attention=False,
|
147 |
apply_quantization=True,
|
148 |
),
|
149 |
+
"gemma-3n-e4b-it": load_model(
|
150 |
+
model_name="google/gemma-3n-e4b-it",
|
151 |
+
use_flash_attention=False,
|
152 |
+
apply_quantization=True,
|
153 |
+
),
|
154 |
}
|
155 |
|
156 |
PROCESSORS = {
|
|
|
160 |
"InternVL3-1B-hf": load_processor("OpenGVLab/InternVL3-1B-hf"),
|
161 |
"InternVL3-2B-hf": load_processor("OpenGVLab/InternVL3-2B-hf"),
|
162 |
"InternVL3-8B-hf": load_processor("OpenGVLab/InternVL3-8B-hf"),
|
163 |
+
"gemma-3n-e4b-it": load_processor("google/gemma-3n-e4b-it"),
|
164 |
}
|
165 |
logger.debug("Models and Processors Loaded!")
|
166 |
|