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

Update app.py

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