:lock: restrict models to 15B or less.
Browse files
app.py
CHANGED
@@ -40,6 +40,8 @@ with open('./model-cache.json', 'r') as f_model_cache:
|
|
40 |
def build_model_choices():
|
41 |
all_choices = []
|
42 |
for model_class in model_cache:
|
|
|
|
|
43 |
all_choices += [ (f"{model_id} ({model_class})", model_id) for model_id in model_cache[model_class] ]
|
44 |
|
45 |
return all_choices
|
|
|
40 |
def build_model_choices():
|
41 |
all_choices = []
|
42 |
for model_class in model_cache:
|
43 |
+
if model_class in ['llama3-70b-8k', 'qwen2-72b-lc']:
|
44 |
+
continue
|
45 |
all_choices += [ (f"{model_id} ({model_class})", model_id) for model_id in model_cache[model_class] ]
|
46 |
|
47 |
return all_choices
|