Spaces:
Running
on
Zero
Running
on
Zero
Update config.py
#3
by
yichuan-huang
- opened
config.py
CHANGED
|
@@ -6,7 +6,7 @@ from dataclasses import dataclass
|
|
| 6 |
@dataclass
|
| 7 |
class Config:
|
| 8 |
# Gemma3n model configuration
|
| 9 |
-
MODEL_NAME: str = "google/gemma-3n-
|
| 10 |
|
| 11 |
# Generation parameters
|
| 12 |
MAX_NEW_TOKENS: int = 512
|
|
|
|
| 6 |
@dataclass
|
| 7 |
class Config:
|
| 8 |
# Gemma3n model configuration
|
| 9 |
+
MODEL_NAME: str = "google/gemma-3n-E4B-it"
|
| 10 |
|
| 11 |
# Generation parameters
|
| 12 |
MAX_NEW_TOKENS: int = 512
|