Spaces:
Running
on
Zero
Running
on
Zero
vinesmsuic
commited on
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 |
|