wxgeorge commited on
Commit
5ae724e
1 Parent(s): fc7d8b9

:lock: restrict models to 15B or less.

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