saumya-pailwan commited on
Commit
3f9bdfa
·
verified ·
1 Parent(s): 640a77c

dropped some models

Browse files
Files changed (1) hide show
  1. vampnet/interface.py +3 -1
vampnet/interface.py CHANGED
@@ -128,7 +128,9 @@ class Interface(torch.nn.Module):
128
  @classmethod
129
  def available_models(cls):
130
  from . import list_finetuned
131
- return list_finetuned() + ["default"]
 
 
132
 
133
 
134
  def load_finetuned(self, name: str):
 
128
  @classmethod
129
  def available_models(cls):
130
  from . import list_finetuned
131
+ all_models = list_finetuned() + ["default"]
132
+ blocked = {"lazaro-ros", "march-31", "sax-new", "sax-sep", "saxophone"}
133
+ return [m for m in all_models if m not in blocked]
134
 
135
 
136
  def load_finetuned(self, name: str):