Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -53,7 +53,6 @@ BASE_MODEL_IDS: List[str] = [
|
|
| 53 |
"OpenCerebral/Boris-1.3-125M",
|
| 54 |
"OpenCerebral/Boris-1.3-75M",
|
| 55 |
"CodeSoft/sorbet-v2-25m",
|
| 56 |
-
"DALabCommunity/Haidass1.5-143M",
|
| 57 |
]
|
| 58 |
|
| 59 |
BASE_MODEL_DISPLAY: Dict[str, str] = {
|
|
@@ -64,7 +63,6 @@ BASE_MODEL_DISPLAY: Dict[str, str] = {
|
|
| 64 |
"OpenCerebral/Boris-1.3-125M": "Boris-1.3-125M",
|
| 65 |
"OpenCerebral/Boris-1.3-75M": "Boris-1.3-75M",
|
| 66 |
"CodeSoft/sorbet-v2-25m": "sorbet-v2-25m",
|
| 67 |
-
"DALabCommunity/Haidass1.5-143M": "Haidass1.5-143M",
|
| 68 |
}
|
| 69 |
|
| 70 |
MODEL_PARAMS: Dict[str, float] = {
|
|
@@ -81,7 +79,6 @@ MODEL_PARAMS: Dict[str, float] = {
|
|
| 81 |
"OpenCerebral/Boris-1.3-125M": 125.0e6,
|
| 82 |
"OpenCerebral/Boris-1.3-75M": 75.0e6,
|
| 83 |
"CodeSoft/sorbet-v2-25m": 25.2e6,
|
| 84 |
-
"DALabCommunity/Haidass1.5-143M": 143.0e6,
|
| 85 |
}
|
| 86 |
|
| 87 |
FALLBACK_IDS: Dict[str, str] = {}
|
|
@@ -165,7 +162,6 @@ GEN_DEFAULTS: Dict[str, dict] = {
|
|
| 165 |
"OpenCerebral/Boris-1.3-125M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 166 |
"OpenCerebral/Boris-1.3-75M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 167 |
"CodeSoft/sorbet-v2-25m": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 168 |
-
"DALabCommunity/Haidass1.5-143M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 169 |
}
|
| 170 |
|
| 171 |
MODEL_CONTEXT: Dict[str, int] = {
|
|
@@ -182,7 +178,6 @@ MODEL_CONTEXT: Dict[str, int] = {
|
|
| 182 |
"OpenCerebral/Boris-1.3-125M": 2048,
|
| 183 |
"OpenCerebral/Boris-1.3-75M": 2048,
|
| 184 |
"CodeSoft/sorbet-v2-25m": 4096,
|
| 185 |
-
"DALabCommunity/Haidass1.5-143M": 4096,
|
| 186 |
}
|
| 187 |
|
| 188 |
|
|
|
|
| 53 |
"OpenCerebral/Boris-1.3-125M",
|
| 54 |
"OpenCerebral/Boris-1.3-75M",
|
| 55 |
"CodeSoft/sorbet-v2-25m",
|
|
|
|
| 56 |
]
|
| 57 |
|
| 58 |
BASE_MODEL_DISPLAY: Dict[str, str] = {
|
|
|
|
| 63 |
"OpenCerebral/Boris-1.3-125M": "Boris-1.3-125M",
|
| 64 |
"OpenCerebral/Boris-1.3-75M": "Boris-1.3-75M",
|
| 65 |
"CodeSoft/sorbet-v2-25m": "sorbet-v2-25m",
|
|
|
|
| 66 |
}
|
| 67 |
|
| 68 |
MODEL_PARAMS: Dict[str, float] = {
|
|
|
|
| 79 |
"OpenCerebral/Boris-1.3-125M": 125.0e6,
|
| 80 |
"OpenCerebral/Boris-1.3-75M": 75.0e6,
|
| 81 |
"CodeSoft/sorbet-v2-25m": 25.2e6,
|
|
|
|
| 82 |
}
|
| 83 |
|
| 84 |
FALLBACK_IDS: Dict[str, str] = {}
|
|
|
|
| 162 |
"OpenCerebral/Boris-1.3-125M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 163 |
"OpenCerebral/Boris-1.3-75M": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
| 164 |
"CodeSoft/sorbet-v2-25m": {"max_new_tokens": 64, "temperature": 0.8, "top_p": 0.95, "repetition_penalty": 1.1, "do_sample": True},
|
|
|
|
| 165 |
}
|
| 166 |
|
| 167 |
MODEL_CONTEXT: Dict[str, int] = {
|
|
|
|
| 178 |
"OpenCerebral/Boris-1.3-125M": 2048,
|
| 179 |
"OpenCerebral/Boris-1.3-75M": 2048,
|
| 180 |
"CodeSoft/sorbet-v2-25m": 4096,
|
|
|
|
| 181 |
}
|
| 182 |
|
| 183 |
|