Update app.py

app.py

@@ -34,6 +34,9 @@ MODEL_IDS: List[str] = [
     "HuggingFaceTB/SmolLM2-135M-Instruct",
     "OpenCerebral/Boris-1.3-125M-Instruct",
     "OpenCerebral/Boris-1.3-75M-Instruct",
+    "BananaMind/BananaMind-2-Pro-Preview-Chat",
+    "BananaMind/BananaMind-2-Nano-Chat",
+    "BananaMind/BananaMind-2-Mini-Chat",
 ]
 
 MODEL_DISPLAY: Dict[str, str] = {
@@ -43,6 +46,9 @@ MODEL_DISPLAY: Dict[str, str] = {
     "HuggingFaceTB/SmolLM2-135M-Instruct": "SmolLM2-135M-Instruct",
     "OpenCerebral/Boris-1.3-125M-Instruct": "Boris-1.3-125M-Instruct",
     "OpenCerebral/Boris-1.3-75M-Instruct": "Boris-1.3-75M-Instruct",
+    "BananaMind/BananaMind-2-Pro-Preview-Chat": "BananaMind-2-Pro-Preview-Chat",
+    "BananaMind/BananaMind-2-Nano-Chat": "BananaMind-2-Nano-Chat",
+    "BananaMind/BananaMind-2-Mini-Chat": "BananaMind-2-Mini-Chat",
 }
 
 BASE_MODEL_IDS: List[str] = [
@@ -56,6 +62,9 @@ BASE_MODEL_IDS: List[str] = [
     "fromziro/Er-Large-30M",
     "fromziro/Negative-v1.0",
     "fromziro/Syn-2.6M",
+    "BananaMind/BananaMind-2-Medium",
+    "BananaMind/BananaMind-2-Mini",
+    "BananaMind/BananaMind-2-Nano",
 ]
 
 BASE_MODEL_DISPLAY: Dict[str, str] = {
@@ -69,6 +78,9 @@ BASE_MODEL_DISPLAY: Dict[str, str] = {
     "fromziro/Er-Large-30M": "Er-Large-30M",
     "fromziro/Negative-v1.0": "Negative-v1.0",
     "fromziro/Syn-2.6M": "Syn-2.6M",
+    "BananaMind/BananaMind-2-Medium": "BananaMind-2-Medium",
+    "BananaMind/BananaMind-2-Mini": "BananaMind-2-Mini",
+    "BananaMind/BananaMind-2-Nano": "BananaMind-2-Nano",
 }
 
 MODEL_PARAMS: Dict[str, float] = {
@@ -88,6 +100,12 @@ MODEL_PARAMS: Dict[str, float] = {
     "fromziro/Er-Large-30M": 31.9e6,
     "fromziro/Negative-v1.0": 67.7e3,
     "fromziro/Syn-2.6M": 2.6e6,
+    "BananaMind/BananaMind-2-Pro-Preview-Chat": 139.0e6,
+    "BananaMind/BananaMind-2-Nano-Chat": 9.97e6,
+    "BananaMind/BananaMind-2-Mini-Chat": 25.18e6,
+    "BananaMind/BananaMind-2-Medium": 55.85e6,
+    "BananaMind/BananaMind-2-Mini": 28.32e6,
+    "BananaMind/BananaMind-2-Nano": 12.07e6,
 }
 
 FALLBACK_IDS: Dict[str, str] = {}
@@ -174,6 +192,12 @@ GEN_DEFAULTS: Dict[str, dict] = {
     "fromziro/Er-Large-30M": {"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},
     "fromziro/Syn-2.6M": {"max_new_tokens": 48, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
+    "BananaMind/BananaMind-2-Pro-Preview-Chat": {"max_new_tokens": 64, "temperature": 0.7, "top_p": 0.9, "repetition_penalty": 1.1, "do_sample": True},
+    "BananaMind/BananaMind-2-Nano-Chat": {"max_new_tokens": 64, "temperature": 0.7, "top_p": 0.9, "repetition_penalty": 1.1, "do_sample": True},
+    "BananaMind/BananaMind-2-Mini-Chat": {"max_new_tokens": 64, "temperature": 0.7, "top_p": 0.9, "repetition_penalty": 1.1, "do_sample": True},
+    "BananaMind/BananaMind-2-Medium": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
+    "BananaMind/BananaMind-2-Mini": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
+    "BananaMind/BananaMind-2-Nano": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
 }
 
 MODEL_CONTEXT: Dict[str, int] = {
@@ -193,6 +217,12 @@ MODEL_CONTEXT: Dict[str, int] = {
     "fromziro/Er-Large-30M": 2048,
     "fromziro/Negative-v1.0": 96,
     "fromziro/Syn-2.6M": 384,
+    "BananaMind/BananaMind-2-Pro-Preview-Chat": 3072,
+    "BananaMind/BananaMind-2-Nano-Chat": 4096,
+    "BananaMind/BananaMind-2-Mini-Chat": 4096,
+    "BananaMind/BananaMind-2-Medium": 3072,
+    "BananaMind/BananaMind-2-Mini": 4096,
+    "BananaMind/BananaMind-2-Nano": 4096,
 }