Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Community download] Fix cache dir (#1555) 

* [Community download] Fix cache dir

* up
This commit is contained in:
Patrick von Platen 2022-12-05 18:52:55 +01:00 committed by GitHub
parent 48d0123f0f
commit ae854746ab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/diffusers/pipeline_utils.py
View File

@ -479,6 +479,7 @@ class DiffusionPipeline(ConfigMixin):
user_agent = {"pipeline_class": requested_pipeline_class}
if custom_pipeline is not None:
user_agent["custom_pipeline"] = custom_pipeline
user_agent = http_user_agent(user_agent)
if is_safetensors_available():
@ -519,9 +520,7 @@ class DiffusionPipeline(ConfigMixin):
else:
file_name = CUSTOM_PIPELINE_FILE_NAME
pipeline_class = get_class_from_dynamic_module(
custom_pipeline, module_file=file_name, cache_dir=custom_pipeline
)
pipeline_class = get_class_from_dynamic_module(custom_pipeline, module_file=file_name, cache_dir=cache_dir)
elif cls != DiffusionPipeline:
pipeline_class = cls
else: