vinesmsuic commited on
Commit
7757ac2
1 Parent(s): 2ced78d

fix model gone problem

Browse files
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