Skip to content

GitLab

Unverified Commit ae854746 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Community download] Fix cache dir (#1555) 

* [Community download] Fix cache dir

* up
parent 48d0123f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
src/diffusers/pipeline_utils.py
View file @ ae854746
...@@ -479,6 +479,7 @@ class DiffusionPipeline(ConfigMixin): ...@@ -479,6 +479,7 @@ class DiffusionPipeline(ConfigMixin):
user_agent = {"pipeline_class": requested_pipeline_class} user_agent = {"pipeline_class": requested_pipeline_class}
if custom_pipeline is not None: if custom_pipeline is not None:
user_agent["custom_pipeline"] = custom_pipeline user_agent["custom_pipeline"] = custom_pipeline
user_agent = http_user_agent(user_agent) user_agent = http_user_agent(user_agent)
if is_safetensors_available(): if is_safetensors_available():
...@@ -519,9 +520,7 @@ class DiffusionPipeline(ConfigMixin): ...@@ -519,9 +520,7 @@ class DiffusionPipeline(ConfigMixin):
else: else:
file_name = CUSTOM_PIPELINE_FILE_NAME file_name = CUSTOM_PIPELINE_FILE_NAME
pipeline_class = get_class_from_dynamic_module( pipeline_class = get_class_from_dynamic_module(custom_pipeline, module_file=file_name, cache_dir=cache_dir)
custom_pipeline, module_file=file_name, cache_dir=custom_pipeline
)
elif cls != DiffusionPipeline: elif cls != DiffusionPipeline:
pipeline_class = cls pipeline_class = cls
else: else:
......
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