Skip to content

Commit 2687b96

Browse files
ahnjjstevhliu
andauthored
Update src/diffusers/models/adapter.py
Co-authored-by: Steven Liu <[email protected]>
1 parent 2db9504 commit 2687b96

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/diffusers/models/adapter.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,8 +123,7 @@ def save_pretrained(
123123
Useful when in distributed training (e.g., TPUs) and need to call this function on all processes.
124124
In this case, set `is_main_process=True` only for the main process to avoid race conditions.
125125
save_function (`Callable`):
126-
Function used to save the state dictionary. Useful on distributed training (e.g., TPUs) when one
127-
need to replace `torch.save` with another method. Can also be configured using`DIFFUSERS_SAVE_MODE` environment variable.
126+
Function used to save the state dictionary. Useful for distributed training (e.g., TPUs) to replace `torch.save` with another method. Can also be configured using`DIFFUSERS_SAVE_MODE` environment variable.
128127
safe_serialization (`bool`, optional, defaults=True):
129128
If `True`, save the model using `safetensors`.
130129
If `False`, save the model with `pickle`.

0 commit comments

Comments
 (0)