Spaces:
Running on Zero
Running on Zero
dropped some models
Browse files- 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 |
-
|
|
|
|
|
|
|
| 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):
|