Skip to content

Commit ae85474

Browse files
[Community download] Fix cache dir (huggingface#1555)
* [Community download] Fix cache dir * up
1 parent 48d0123 commit ae85474

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/diffusers/pipeline_utils.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -479,6 +479,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
479479
user_agent = {"pipeline_class": requested_pipeline_class}
480480
if custom_pipeline is not None:
481481
user_agent["custom_pipeline"] = custom_pipeline
482+
482483
user_agent = http_user_agent(user_agent)
483484

484485
if is_safetensors_available():
@@ -519,9 +520,7 @@ def from_pretrained(cls, pretrained_model_name_or_path: Optional[Union[str, os.P
519520
else:
520521
file_name = CUSTOM_PIPELINE_FILE_NAME
521522

522-
pipeline_class = get_class_from_dynamic_module(
523-
custom_pipeline, module_file=file_name, cache_dir=custom_pipeline
524-
)
523+
pipeline_class = get_class_from_dynamic_module(custom_pipeline, module_file=file_name, cache_dir=cache_dir)
525524
elif cls != DiffusionPipeline:
526525
pipeline_class = cls
527526
else:

0 commit comments

Comments
 (0)