Update config.py (#3)

- Update config.py (35e8da89535901f369d200182521a3e22a49fd47)


Co-authored-by: Yichuan Huang <yichuan-huang@users.noreply.huggingface.co>

config.py

@@ -6,7 +6,7 @@ from dataclasses import dataclass
 @dataclass
 class Config:
     # Gemma3n model configuration
-    MODEL_NAME: str = "google/gemma-3n-E2B-it"
+    MODEL_NAME: str = "google/gemma-3n-E4B-it"
 
     # Generation parameters
     MAX_NEW_TOKENS: int = 512