[Tests] Move stable diffusion into their own files (#936)

* [Tests] Move stable diffusion into their own files * up
huggingface · Oct 21, 2022 · 25dfd0f · 25dfd0f
1 parent 32bf4fd
commit 25dfd0f
Show file tree

Hide file tree

Showing 6 changed files with 2,283 additions and 1,680 deletions.
diff --git a/src/diffusers/pipeline_utils.py b/src/diffusers/pipeline_utils.py
@@ -122,7 +122,7 @@ def register_modules(self, **kwargs):
                 library = module.__module__.split(".")[0]
 
                 # check if the module is a pipeline module
-                pipeline_dir = module.__module__.split(".")[-2]
+                pipeline_dir = module.__module__.split(".")[-2] if len(module.__module__.split(".")) > 2 else None
                 path = module.__module__.split(".")
                 is_pipeline_module = pipeline_dir in path and hasattr(pipelines, pipeline_dir)