Update dataclasses.py (#1894)

huggingface · Aug 28, 2023 · 1ace672 · 1ace672
1 parent e2ae254
commit 1ace672
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/accelerate/utils/dataclasses.py b/src/accelerate/utils/dataclasses.py
@@ -725,7 +725,7 @@ def set_deepspeed_weakref(self):
             if ds_config["train_batch_size"] == "auto":
                 del ds_config["train_batch_size"]
 
-            from transformers.deepspeed import HfDeepSpeedConfig
+            from transformers.integrations import HfDeepSpeedConfig
 
             self.dschf = HfDeepSpeedConfig(ds_config)  # keep this object alive # noqa