CodeSoft commited on
Commit
c9ca161
·
verified ·
1 Parent(s): cecd7fd

Update app.py

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