Skip to content

Commit fa15369

Browse files
[bugfix] Check that model_index.json module is in required_modules list before removing (#756)
1 parent 78a9cb8 commit fa15369

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

fastvideo/pipelines/composed_pipeline_base.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -282,7 +282,11 @@ def load_modules(
282282
for module_name, (transformers_or_diffusers,
283283
architecture) in model_index.items():
284284
if transformers_or_diffusers is None:
285-
self.required_config_modules.remove(module_name)
285+
logger.warning(
286+
"Module in model_index.json has null value, removing from required_config_modules"
287+
)
288+
if module_name in self.required_config_modules:
289+
self.required_config_modules.remove(module_name)
286290
continue
287291
if module_name not in required_modules:
288292
logger.info("Skipping module %s", module_name)

0 commit comments

Comments
 (0)