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

* [Tests] Move stable diffusion into their own files * up
yoonseokjin · Oct 21, 2022 · ffc5dd6 · ffc5dd6
1 parent 100310d
commit ffc5dd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pipeline_utils.py b/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)