Skip to content

Commit f0d386f

Browse files
authored
Merge branch 'main' into Notebooks-for-Community-Scripts
2 parents af7c3ef + c7a8c43 commit f0d386f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1462,6 +1462,8 @@ def download(cls, pretrained_model_name, **kwargs) -> Union[str, os.PathLike]:
14621462
allow_patterns += [f"{custom_pipeline}.py"] if f"{custom_pipeline}.py" in filenames else []
14631463
# also allow downloading config.json files with the model
14641464
allow_patterns += [os.path.join(k, "config.json") for k in model_folder_names]
1465+
# also allow downloading generation_config.json of the transformers model
1466+
allow_patterns += [os.path.join(k, "generation_config.json") for k in model_folder_names]
14651467
allow_patterns += [
14661468
SCHEDULER_CONFIG_NAME,
14671469
CONFIG_NAME,

0 commit comments

Comments
 (0)