fix how is_fsdp is determined (#12960)

sayakpaul · web-flow · commit d44b5f86e613 · 2026-01-10T10:34:25.000+05:30
up
diff --git a/examples/dreambooth/train_dreambooth_lora_flux2.py b/examples/dreambooth/train_dreambooth_lora_flux2.py
@@ -1228,7 +1228,7 @@ def main(args):
         else {"device": accelerator.device, "dtype": weight_dtype}
     )
 
-    is_fsdp = accelerator.state.fsdp_plugin is not None
+    is_fsdp = getattr(accelerator.state, "fsdp_plugin", None) is not None
     if not is_fsdp:
         transformer.to(**transformer_to_kwargs)
 
diff --git a/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py b/examples/dreambooth/train_dreambooth_lora_flux2_img2img.py
@@ -1178,7 +1178,7 @@ def main(args):
         else {"device": accelerator.device, "dtype": weight_dtype}
     )
 
-    is_fsdp = accelerator.state.fsdp_plugin is not None
+    is_fsdp = getattr(accelerator.state, "fsdp_plugin", None) is not None
     if not is_fsdp:
         transformer.to(**transformer_to_kwargs)
 

Original file line number	Diff line number	Diff line change
`@@ -1228,7 +1228,7 @@ def main(args):`
`1228`	`1228`	`else {"device": accelerator.device, "dtype": weight_dtype}`
`1229`	`1229`	`)`
`1230`	`1230`
`1231`		`- is_fsdp = accelerator.state.fsdp_plugin is not None`
	`1231`	`+ is_fsdp = getattr(accelerator.state, "fsdp_plugin", None) is not None`
`1232`	`1232`	`if not is_fsdp:`
`1233`	`1233`	`transformer.to(**transformer_to_kwargs)`
`1234`	`1234`
Original file line number	Diff line number	Diff line change
`@@ -1178,7 +1178,7 @@ def main(args):`
`1178`	`1178`	`else {"device": accelerator.device, "dtype": weight_dtype}`
`1179`	`1179`	`)`
`1180`	`1180`
`1181`		`- is_fsdp = accelerator.state.fsdp_plugin is not None`
	`1181`	`+ is_fsdp = getattr(accelerator.state, "fsdp_plugin", None) is not None`
`1182`	`1182`	`if not is_fsdp:`
`1183`	`1183`	`transformer.to(**transformer_to_kwargs)`
`1184`	`1184`