Update app.py

app.py

@@ -69,6 +69,9 @@ BASE_MODEL_IDS: List[str] = [
     "BananaMind/BananaMind-2-Medium",
     "BananaMind/BananaMind-2-Mini",
     "BananaMind/BananaMind-2-Nano",
+    "DedeProGames/Kiyo-230M-Preview",
+    "DedeProGames/Kiyo-135M",
+    "DedeProGames/Kiyo-65M",
 ]
 
 BASE_MODEL_DISPLAY: Dict[str, str] = {
@@ -85,6 +88,9 @@ BASE_MODEL_DISPLAY: Dict[str, str] = {
     "BananaMind/BananaMind-2-Medium": "BananaMind-2-Medium",
     "BananaMind/BananaMind-2-Mini": "BananaMind-2-Mini",
     "BananaMind/BananaMind-2-Nano": "BananaMind-2-Nano",
+    "DedeProGames/Kiyo-230M-Preview": "Kiyo-230M-Preview",
+    "DedeProGames/Kiyo-135M": "Kiyo-135M",
+    "DedeProGames/Kiyo-65M": "Kiyo-65M",
 }
 
 MODEL_PARAMS: Dict[str, float] = {
@@ -112,6 +118,9 @@ MODEL_PARAMS: Dict[str, float] = {
     "BananaMind/BananaMind-2-Medium": 55.85e6,
     "BananaMind/BananaMind-2-Mini": 28.32e6,
     "BananaMind/BananaMind-2-Nano": 12.07e6,
+    "DedeProGames/Kiyo-230M-Preview": 229.7e6,
+    "DedeProGames/Kiyo-135M": 134.5e6,
+    "DedeProGames/Kiyo-65M": 65.0e6,
 }
 
 FALLBACK_IDS: Dict[str, str] = {}
@@ -206,6 +215,9 @@ GEN_DEFAULTS: Dict[str, dict] = {
     "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},
+    "DedeProGames/Kiyo-230M-Preview": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
+    "DedeProGames/Kiyo-135M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
+    "DedeProGames/Kiyo-65M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
 }
 
 MODEL_CONTEXT: Dict[str, int] = {
@@ -233,6 +245,9 @@ MODEL_CONTEXT: Dict[str, int] = {
     "BananaMind/BananaMind-2-Medium": 3072,
     "BananaMind/BananaMind-2-Mini": 4096,
     "BananaMind/BananaMind-2-Nano": 4096,
+    "DedeProGames/Kiyo-230M-Preview": 2048,
+    "DedeProGames/Kiyo-135M": 8192,
+    "DedeProGames/Kiyo-65M": 2048,
 }