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 |
|