Commit 109d5bbe authored by Patrick von Platen's avatar Patrick von Platen
Browse files
parents f277d5e5 28e8d1f6
...@@ -34,8 +34,11 @@ def filter_pipelines(usage_dict, usage_cutoff=10000): ...@@ -34,8 +34,11 @@ def filter_pipelines(usage_dict, usage_cutoff=10000):
if usage < usage_cutoff: if usage < usage_cutoff:
continue continue
if "Pipeline" in diffusers_object: is_diffusers_pipeline = hasattr(diffusers.pipelines, diffusers_object)
output.append(diffusers_object) if not is_diffusers_pipeline:
continue
output.append(diffusers_object)
return output return output
...@@ -71,6 +74,7 @@ def fetch_pipeline_modules_to_test(): ...@@ -71,6 +74,7 @@ def fetch_pipeline_modules_to_test():
test_modules = [] test_modules = []
for pipeline_name in pipeline_objects: for pipeline_name in pipeline_objects:
module = getattr(diffusers, pipeline_name) module = getattr(diffusers, pipeline_name)
test_module = module.__module__.split(".")[-2].strip() test_module = module.__module__.split(".")[-2].strip()
test_modules.append(test_module) test_modules.append(test_module)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment