We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1a5797c commit 3a237f4Copy full SHA for 3a237f4
examples/text_to_image/train_text_to_image.py
@@ -471,7 +471,7 @@ def deepspeed_zero_init_disabled_context_manager():
471
"""
472
returns either a context list that includes one that will disable zero.Init or an empty context list
473
474
- deepspeed_plugin = AcceleratorState() if accelerate.state.is_initialized() else None
+ deepspeed_plugin = AcceleratorState().deepspeed_plugin if accelerate.state.is_initialized() else None
475
if deepspeed_plugin is None:
476
return []
477
0 commit comments