AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

up

Browse Source
This commit is contained in:
sayakpaul
2026-01-12 13:47:32 +05:30
parent c13b264589
commit 6983485eed
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/training_utils.py
View File

@@ -432,7 +432,7 @@ def get_fsdp_kwargs_from_accelerator(accelerator) -> dict:
def wrap_with_fsdp(
model: torch.nn.Module,
device: str | torch.Device,
device: str | torch.device,
offload: bool = True,
use_orig_params: bool = True,
limit_all_gathers: bool = True,