saumya-pailwan commited on
Commit
3b90a37
·
verified ·
1 Parent(s): d5e1664

filtered models fix

Browse files
Files changed (1) hide show
  1. vampnet/interface.py +3 -4
vampnet/interface.py CHANGED
@@ -129,10 +129,9 @@ class Interface(torch.nn.Module):
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
- print("Using filtered models:", [m for m in all_models if m not in blocked])
134
- logging.debug("Using filtered models:", [m for m in all_models if m not in blocked])
135
- return [m for m in all_models if m not in blocked]
136
 
137
 
138
  def load_finetuned(self, name: str):
 
129
  def available_models(cls):
130
  from . import list_finetuned
131
  all_models = list_finetuned() + ["default"]
132
+ filtered = ["default", "cat10", "lazaro-ros-sep", "supermetroid"]
133
+ print("Using filtered models:", filtered)
134
+ return filtered
 
135
 
136
 
137
  def load_finetuned(self, name: str):