You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: src/diffusers/models/adapter.py
+1-2Lines changed: 1 addition & 2 deletions
Original file line number
Diff line number
Diff line change
@@ -123,8 +123,7 @@ def save_pretrained(
123
123
Useful when in distributed training (e.g., TPUs) and need to call this function on all processes.
124
124
In this case, set `is_main_process=True` only for the main process to avoid race conditions.
125
125
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.
0 commit comments