Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
7757ac2
1
Parent(s):
2ced78d
fix model gone problem
Browse files- model/model_manager.py +1 -0
- model/models/__init__.py +1 -1
model/model_manager.py
CHANGED
|
@@ -195,6 +195,7 @@ class ModelManager:
|
|
| 195 |
def generate_video_vg_museum_parallel_anony(self, model_A, model_B):
|
| 196 |
# Using list comprehension to get the difference between two lists
|
| 197 |
picking_list = [item for item in self.model_vg_list if item not in self.excluding_model_list]
|
|
|
|
| 198 |
if model_A == "" and model_B == "":
|
| 199 |
model_names = random.sample([model for model in picking_list], 2)
|
| 200 |
|
|
|
|
| 195 |
def generate_video_vg_museum_parallel_anony(self, model_A, model_B):
|
| 196 |
# Using list comprehension to get the difference between two lists
|
| 197 |
picking_list = [item for item in self.model_vg_list if item not in self.excluding_model_list]
|
| 198 |
+
picking_list = [item for item in picking_list if item not in self.desired_model_list]
|
| 199 |
if model_A == "" and model_B == "":
|
| 200 |
model_names = random.sample([model for model in picking_list], 2)
|
| 201 |
|
model/models/__init__.py
CHANGED
|
@@ -19,7 +19,7 @@ VIDEO_GENERATION_MODELS = ['fal_AnimateDiff_text2video',
|
|
| 19 |
'videogenhub_LaVie_generation',
|
| 20 |
'videogenhub_VideoCrafter2_generation',
|
| 21 |
'videogenhub_ModelScope_generation',
|
| 22 |
-
'videogenhub_OpenSora_generation']
|
| 23 |
MUSEUM_UNSUPPORTED_MODELS = ['videogenhub_OpenSoraPlan_generation']
|
| 24 |
DESIRED_APPEAR_MODEL = ['videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
|
| 25 |
|
|
|
|
| 19 |
'videogenhub_LaVie_generation',
|
| 20 |
'videogenhub_VideoCrafter2_generation',
|
| 21 |
'videogenhub_ModelScope_generation',
|
| 22 |
+
'videogenhub_OpenSora_generation', 'videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
|
| 23 |
MUSEUM_UNSUPPORTED_MODELS = ['videogenhub_OpenSoraPlan_generation']
|
| 24 |
DESIRED_APPEAR_MODEL = ['videogenhub_T2VTurbo_generation','fal_StableVideoDiffusion_text2video']
|
| 25 |
|