CodeSoft commited on
Commit
2d6cc43
·
verified ·
1 Parent(s): 1f74c4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
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