Update app.py

app.py

@@ -53,6 +53,9 @@ BASE_MODEL_IDS: List[str] = [
     "OpenCerebral/Boris-1.3-125M",
     "OpenCerebral/Boris-1.3-75M",
     "CodeSoft/sorbet-v2-25m",
+    "fromziro/Er-Large-31M",
+    "fromziro/Negative-v1.0",
+    "FromZero/Syn-2.6M",
 ]
 
 BASE_MODEL_DISPLAY: Dict[str, str] = {
@@ -63,6 +66,9 @@ BASE_MODEL_DISPLAY: Dict[str, str] = {
     "OpenCerebral/Boris-1.3-125M": "Boris-1.3-125M",
     "OpenCerebral/Boris-1.3-75M": "Boris-1.3-75M",
     "CodeSoft/sorbet-v2-25m": "sorbet-v2-25m",
+    "fromziro/Er-Large-31M": "Er-Large-31M",
+    "fromziro/Negative-v1.0": "Negative-v1.0",
+    "FromZero/Syn-2.6M": "Syn-2.6M",
 }
 
 MODEL_PARAMS: Dict[str, float] = {
@@ -79,6 +85,9 @@ MODEL_PARAMS: Dict[str, float] = {
     "OpenCerebral/Boris-1.3-125M": 125.0e6,
     "OpenCerebral/Boris-1.3-75M": 75.0e6,
     "CodeSoft/sorbet-v2-25m": 25.2e6,
+    "fromziro/Er-Large-31M": 31.9e6,
+    "fromziro/Negative-v1.0": 67.7e3,
+    "FromZero/Syn-2.6M": 2.6e6,
 }
 
 FALLBACK_IDS: Dict[str, str] = {}
@@ -162,6 +171,9 @@ GEN_DEFAULTS: Dict[str, dict] = {
     "OpenCerebral/Boris-1.3-125M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
     "OpenCerebral/Boris-1.3-75M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
     "CodeSoft/sorbet-v2-25m": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
+    "fromziro/Er-Large-31M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
+    "fromziro/Negative-v1.0": {"max_new_tokens": 48, "temperature": 0.9, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
+    "FromZero/Syn-2.6M": {"max_new_tokens": 48, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
 }
 
 MODEL_CONTEXT: Dict[str, int] = {
@@ -178,6 +190,9 @@ MODEL_CONTEXT: Dict[str, int] = {
     "OpenCerebral/Boris-1.3-125M": 2048,
     "OpenCerebral/Boris-1.3-75M": 2048,
     "CodeSoft/sorbet-v2-25m": 4096,
+    "fromziro/Er-Large-31M": 2048,
+    "fromziro/Negative-v1.0": 96,
+    "FromZero/Syn-2.6M": 384,
 }