Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -71,9 +71,6 @@ BASE_MODEL_IDS: List[str] = [
|
|
| 71 |
"BananaMind/BananaMind-2-Medium",
|
| 72 |
"BananaMind/BananaMind-2-Mini",
|
| 73 |
"BananaMind/BananaMind-2-Nano",
|
| 74 |
-
"DedeProGames/Kiyo-230M-Preview",
|
| 75 |
-
"DedeProGames/Kiyo-135M",
|
| 76 |
-
"DedeProGames/Kiyo-65M",
|
| 77 |
"IvmeLabs/Ivme-Conversate-v2-Base",
|
| 78 |
"IvmeLabs/Ivme-Conversate-v3-Base",
|
| 79 |
]
|
|
@@ -92,9 +89,6 @@ BASE_MODEL_DISPLAY: Dict[str, str] = {
|
|
| 92 |
"BananaMind/BananaMind-2-Medium": "BananaMind-2-Medium",
|
| 93 |
"BananaMind/BananaMind-2-Mini": "BananaMind-2-Mini",
|
| 94 |
"BananaMind/BananaMind-2-Nano": "BananaMind-2-Nano",
|
| 95 |
-
"DedeProGames/Kiyo-230M-Preview": "Kiyo-230M-Preview",
|
| 96 |
-
"DedeProGames/Kiyo-135M": "Kiyo-135M",
|
| 97 |
-
"DedeProGames/Kiyo-65M": "Kiyo-65M",
|
| 98 |
"IvmeLabs/Ivme-Conversate-v2-Base": "Ivme-Conversate-v2-Base",
|
| 99 |
"IvmeLabs/Ivme-Conversate-v3-Base": "Ivme-Conversate-v3-Base",
|
| 100 |
}
|
|
@@ -124,9 +118,6 @@ MODEL_PARAMS: Dict[str, float] = {
|
|
| 124 |
"BananaMind/BananaMind-2-Medium": 55.85e6,
|
| 125 |
"BananaMind/BananaMind-2-Mini": 28.32e6,
|
| 126 |
"BananaMind/BananaMind-2-Nano": 12.07e6,
|
| 127 |
-
"DedeProGames/Kiyo-230M-Preview": 229.7e6,
|
| 128 |
-
"DedeProGames/Kiyo-135M": 134.5e6,
|
| 129 |
-
"DedeProGames/Kiyo-65M": 65.0e6,
|
| 130 |
"wayneworkman2012/peacebell-v1-148M": 148.55e6,
|
| 131 |
"IvmeLabs/Ivme-Conversate-v2-Base": 23.8e6,
|
| 132 |
"IvmeLabs/Ivme-Conversate-v3-Base": 24.8e6,
|
|
@@ -224,9 +215,6 @@ GEN_DEFAULTS: Dict[str, dict] = {
|
|
| 224 |
"BananaMind/BananaMind-2-Medium": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 225 |
"BananaMind/BananaMind-2-Mini": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 226 |
"BananaMind/BananaMind-2-Nano": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 227 |
-
"DedeProGames/Kiyo-230M-Preview": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 228 |
-
"DedeProGames/Kiyo-135M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 229 |
-
"DedeProGames/Kiyo-65M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 230 |
# Peacebell: greedy (temperature 0), no repetition penalty, no n-gram block; decoded by its own ChatML path (generate_peacebell)
|
| 231 |
"wayneworkman2012/peacebell-v1-148M": {"max_new_tokens": 64, "temperature": 0.0, "do_sample": False, "repetition_penalty": 1.0},
|
| 232 |
"IvmeLabs/Ivme-Conversate-v2-Base": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
|
@@ -258,9 +246,6 @@ MODEL_CONTEXT: Dict[str, int] = {
|
|
| 258 |
"BananaMind/BananaMind-2-Medium": 3072,
|
| 259 |
"BananaMind/BananaMind-2-Mini": 4096,
|
| 260 |
"BananaMind/BananaMind-2-Nano": 4096,
|
| 261 |
-
"DedeProGames/Kiyo-230M-Preview": 2048,
|
| 262 |
-
"DedeProGames/Kiyo-135M": 8192,
|
| 263 |
-
"DedeProGames/Kiyo-65M": 2048,
|
| 264 |
"wayneworkman2012/peacebell-v1-148M": 16384,
|
| 265 |
"IvmeLabs/Ivme-Conversate-v2-Base": 1024,
|
| 266 |
"IvmeLabs/Ivme-Conversate-v3-Base": 1024,
|
|
|
|
| 71 |
"BananaMind/BananaMind-2-Medium",
|
| 72 |
"BananaMind/BananaMind-2-Mini",
|
| 73 |
"BananaMind/BananaMind-2-Nano",
|
|
|
|
|
|
|
|
|
|
| 74 |
"IvmeLabs/Ivme-Conversate-v2-Base",
|
| 75 |
"IvmeLabs/Ivme-Conversate-v3-Base",
|
| 76 |
]
|
|
|
|
| 89 |
"BananaMind/BananaMind-2-Medium": "BananaMind-2-Medium",
|
| 90 |
"BananaMind/BananaMind-2-Mini": "BananaMind-2-Mini",
|
| 91 |
"BananaMind/BananaMind-2-Nano": "BananaMind-2-Nano",
|
|
|
|
|
|
|
|
|
|
| 92 |
"IvmeLabs/Ivme-Conversate-v2-Base": "Ivme-Conversate-v2-Base",
|
| 93 |
"IvmeLabs/Ivme-Conversate-v3-Base": "Ivme-Conversate-v3-Base",
|
| 94 |
}
|
|
|
|
| 118 |
"BananaMind/BananaMind-2-Medium": 55.85e6,
|
| 119 |
"BananaMind/BananaMind-2-Mini": 28.32e6,
|
| 120 |
"BananaMind/BananaMind-2-Nano": 12.07e6,
|
|
|
|
|
|
|
|
|
|
| 121 |
"wayneworkman2012/peacebell-v1-148M": 148.55e6,
|
| 122 |
"IvmeLabs/Ivme-Conversate-v2-Base": 23.8e6,
|
| 123 |
"IvmeLabs/Ivme-Conversate-v3-Base": 24.8e6,
|
|
|
|
| 215 |
"BananaMind/BananaMind-2-Medium": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 216 |
"BananaMind/BananaMind-2-Mini": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 217 |
"BananaMind/BananaMind-2-Nano": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
|
|
|
|
|
|
|
|
|
| 218 |
# Peacebell: greedy (temperature 0), no repetition penalty, no n-gram block; decoded by its own ChatML path (generate_peacebell)
|
| 219 |
"wayneworkman2012/peacebell-v1-148M": {"max_new_tokens": 64, "temperature": 0.0, "do_sample": False, "repetition_penalty": 1.0},
|
| 220 |
"IvmeLabs/Ivme-Conversate-v2-Base": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
|
|
|
| 246 |
"BananaMind/BananaMind-2-Medium": 3072,
|
| 247 |
"BananaMind/BananaMind-2-Mini": 4096,
|
| 248 |
"BananaMind/BananaMind-2-Nano": 4096,
|
|
|
|
|
|
|
|
|
|
| 249 |
"wayneworkman2012/peacebell-v1-148M": 16384,
|
| 250 |
"IvmeLabs/Ivme-Conversate-v2-Base": 1024,
|
| 251 |
"IvmeLabs/Ivme-Conversate-v3-Base": 1024,
|