Skip to content

GitLab

Unverified Commit 3a237f4f authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

fix: deepseepd_plugin retrieval from accelerate state (#3410)

parent 1a5797c6
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/text_to_image/train_text_to_image.py
View file @ 3a237f4f
...@@ -471,7 +471,7 @@ def main(): ...@@ -471,7 +471,7 @@ def main():
""" """
returns either a context list that includes one that will disable zero.Init or an empty context list returns either a context list that includes one that will disable zero.Init or an empty context list
""" """
deepspeed_plugin = AcceleratorState() if accelerate.state.is_initialized() else None deepspeed_plugin = AcceleratorState().deepspeed_plugin if accelerate.state.is_initialized() else None
if deepspeed_plugin is None: if deepspeed_plugin is None:
return [] return []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment