Skip to content

Commit 25dfd0f

Browse files
[Tests] Move stable diffusion into their own files (#936)
* [Tests] Move stable diffusion into their own files * up
1 parent 32bf4fd commit 25dfd0f

File tree

6 files changed

+2283
-1680
lines changed

6 files changed

+2283
-1680
lines changed

src/diffusers/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)