Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -65,7 +65,6 @@ BASE_MODEL_IDS: List[str] = [
|
|
| 65 |
"BananaMind/BananaMind-2-Medium",
|
| 66 |
"BananaMind/BananaMind-2-Mini",
|
| 67 |
"BananaMind/BananaMind-2-Nano",
|
| 68 |
-
"opencerebral/Boris-1.7-D60M-n30M",
|
| 69 |
]
|
| 70 |
|
| 71 |
BASE_MODEL_DISPLAY: Dict[str, str] = {
|
|
@@ -82,7 +81,6 @@ BASE_MODEL_DISPLAY: Dict[str, str] = {
|
|
| 82 |
"BananaMind/BananaMind-2-Medium": "BananaMind-2-Medium",
|
| 83 |
"BananaMind/BananaMind-2-Mini": "BananaMind-2-Mini",
|
| 84 |
"BananaMind/BananaMind-2-Nano": "BananaMind-2-Nano",
|
| 85 |
-
"opencerebral/Boris-1.7-D60M-n30M": "Boris-1.7-D60M-n30M",
|
| 86 |
}
|
| 87 |
|
| 88 |
MODEL_PARAMS: Dict[str, float] = {
|
|
@@ -108,7 +106,6 @@ MODEL_PARAMS: Dict[str, float] = {
|
|
| 108 |
"BananaMind/BananaMind-2-Medium": 55.85e6,
|
| 109 |
"BananaMind/BananaMind-2-Mini": 28.32e6,
|
| 110 |
"BananaMind/BananaMind-2-Nano": 12.07e6,
|
| 111 |
-
"opencerebral/Boris-1.7-D60M-n30M": 116.1e6,
|
| 112 |
}
|
| 113 |
|
| 114 |
FALLBACK_IDS: Dict[str, str] = {}
|
|
@@ -201,7 +198,6 @@ GEN_DEFAULTS: Dict[str, dict] = {
|
|
| 201 |
"BananaMind/BananaMind-2-Medium": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 202 |
"BananaMind/BananaMind-2-Mini": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 203 |
"BananaMind/BananaMind-2-Nano": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 204 |
-
"opencerebral/Boris-1.7-D60M-n30M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 205 |
}
|
| 206 |
|
| 207 |
MODEL_CONTEXT: Dict[str, int] = {
|
|
@@ -227,7 +223,6 @@ MODEL_CONTEXT: Dict[str, int] = {
|
|
| 227 |
"BananaMind/BananaMind-2-Medium": 3072,
|
| 228 |
"BananaMind/BananaMind-2-Mini": 4096,
|
| 229 |
"BananaMind/BananaMind-2-Nano": 4096,
|
| 230 |
-
"opencerebral/Boris-1.7-D60M-n30M": 1024,
|
| 231 |
}
|
| 232 |
|
| 233 |
|
|
@@ -1388,4 +1383,4 @@ if __name__ == "__main__":
|
|
| 1388 |
logger.warning(f"Leaderboard preview failed: {e}")
|
| 1389 |
demo = create_demo()
|
| 1390 |
demo.queue(max_size=20)
|
| 1391 |
-
demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True, theme=gr.themes.Base(), css=CSS)
|
|
|
|
| 65 |
"BananaMind/BananaMind-2-Medium",
|
| 66 |
"BananaMind/BananaMind-2-Mini",
|
| 67 |
"BananaMind/BananaMind-2-Nano",
|
|
|
|
| 68 |
]
|
| 69 |
|
| 70 |
BASE_MODEL_DISPLAY: Dict[str, str] = {
|
|
|
|
| 81 |
"BananaMind/BananaMind-2-Medium": "BananaMind-2-Medium",
|
| 82 |
"BananaMind/BananaMind-2-Mini": "BananaMind-2-Mini",
|
| 83 |
"BananaMind/BananaMind-2-Nano": "BananaMind-2-Nano",
|
|
|
|
| 84 |
}
|
| 85 |
|
| 86 |
MODEL_PARAMS: Dict[str, float] = {
|
|
|
|
| 106 |
"BananaMind/BananaMind-2-Medium": 55.85e6,
|
| 107 |
"BananaMind/BananaMind-2-Mini": 28.32e6,
|
| 108 |
"BananaMind/BananaMind-2-Nano": 12.07e6,
|
|
|
|
| 109 |
}
|
| 110 |
|
| 111 |
FALLBACK_IDS: Dict[str, str] = {}
|
|
|
|
| 198 |
"BananaMind/BananaMind-2-Medium": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 199 |
"BananaMind/BananaMind-2-Mini": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 200 |
"BananaMind/BananaMind-2-Nano": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
|
|
|
| 201 |
}
|
| 202 |
|
| 203 |
MODEL_CONTEXT: Dict[str, int] = {
|
|
|
|
| 223 |
"BananaMind/BananaMind-2-Medium": 3072,
|
| 224 |
"BananaMind/BananaMind-2-Mini": 4096,
|
| 225 |
"BananaMind/BananaMind-2-Nano": 4096,
|
|
|
|
| 226 |
}
|
| 227 |
|
| 228 |
|
|
|
|
| 1383 |
logger.warning(f"Leaderboard preview failed: {e}")
|
| 1384 |
demo = create_demo()
|
| 1385 |
demo.queue(max_size=20)
|
| 1386 |
+
demo.launch(server_name="0.0.0.0", server_port=7860, show_error=True, theme=gr.themes.Base(), css=CSS)
|