From 0a4819a75575bf0d183b544bf92a27eade1868aa Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Mon, 7 Jul 2025 09:49:29 +0200 Subject: [PATCH] add sub_folder to save_pretrained() for config mixin --- src/diffusers/configuration_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/configuration_utils.py b/src/diffusers/configuration_utils.py index 770c949ffb..048ddcae32 100644 --- a/src/diffusers/configuration_utils.py +++ b/src/diffusers/configuration_utils.py @@ -176,6 +176,7 @@ class ConfigMixin: token = kwargs.pop("token", None) repo_id = kwargs.pop("repo_id", save_directory.split(os.path.sep)[-1]) repo_id = create_repo(repo_id, exist_ok=True, private=private, token=token).repo_id + subfolder = kwargs.pop("subfolder", None) self._upload_folder( save_directory, @@ -183,6 +184,7 @@ class ConfigMixin: token=token, commit_message=commit_message, create_pr=create_pr, + subfolder=subfolder, ) @classmethod