Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -65,6 +65,7 @@ BASE_MODEL_IDS: List[str] = [
|
|
| 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,6 +82,7 @@ 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,6 +108,7 @@ 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,6 +201,7 @@ GEN_DEFAULTS: Dict[str, dict] = {
|
|
| 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,6 +227,7 @@ 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 |
|
|
|
|
| 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 |
"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 |
"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 |
"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 |
"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 |
|