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
Indicates whether current process is the main process or not.
123
-
Useful when in distributed training (e.g., TPUs) and need to call this function on all processes.
123
+
Useful for 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
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