Skip to content

Commit ffc5dd6

Browse files
[Tests] Move stable diffusion into their own files (huggingface#936)
* [Tests] Move stable diffusion into their own files * up
1 parent 100310d commit ffc5dd6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pipeline_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -122,7 +122,7 @@ def register_modules(self, **kwargs):
122122
library = module.__module__.split(".")[0]
123123

124124
# check if the module is a pipeline module
125-
pipeline_dir = module.__module__.split(".")[-2]
125+
pipeline_dir = module.__module__.split(".")[-2] if len(module.__module__.split(".")) > 2 else None
126126
path = module.__module__.split(".")
127127
is_pipeline_module = pipeline_dir in path and hasattr(pipelines, pipeline_dir)
128128

0 commit comments

Comments
 (0)