Spaces:
Running
on
Zero
Running
on
Zero
filtered models fix
Browse files- 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 |
-
|
| 133 |
-
print("Using filtered models:",
|
| 134 |
-
|
| 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):
|