John Ho commited on
Commit
5a0f6f1
·
1 Parent(s): 3feaae8

try to load gemma again

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -146,11 +146,11 @@ MODEL_ZOO = {
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,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
- # "gemma-3n-e4b-it": load_processor("google/gemma-3n-e4b-it"),
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