Skip to content

Commit 0a4819a

Browse files
committed
add sub_folder to save_pretrained() for config mixin
1 parent 7cea9a3 commit 0a4819a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/diffusers/configuration_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,13 +176,15 @@ def save_config(self, save_directory: Union[str, os.PathLike], push_to_hub: bool
176176
token = kwargs.pop("token", None)
177177
repo_id = kwargs.pop("repo_id", save_directory.split(os.path.sep)[-1])
178178
repo_id = create_repo(repo_id, exist_ok=True, private=private, token=token).repo_id
179+
subfolder = kwargs.pop("subfolder", None)
179180

180181
self._upload_folder(
181182
save_directory,
182183
repo_id,
183184
token=token,
184185
commit_message=commit_message,
185186
create_pr=create_pr,
187+
subfolder=subfolder,
186188
)
187189

188190
@classmethod

0 commit comments

Comments
 (0)