Import deepspeed utilities from integrations (#25919)

Follow up from #25599

Import deepspeed utilities from integrations (#25919)
Follow up from #25599
bfb1895e · Omar Sanseviero · GitHub · eb984418 · bfb1895e
Unverified Commit bfb1895e authored Sep 04, 2023 by Omar Sanseviero Committed by GitHub Sep 04, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/models/vits/modeling_vits.py src/transformers/models/vits/modeling_vits.py +1 -1

No files found.
--- a/src/transformers/models/vits/modeling_vits.py
+++ b/src/transformers/models/vits/modeling_vits.py
@@ -24,7 +24,7 @@ import torch.utils.checkpoint
 from torch import nn

 from ...activations import ACT2FN
-from ...deepspeed import is_deepspeed_zero3_enabled
+from ...integrations.deepspeed import is_deepspeed_zero3_enabled
 from ...modeling_outputs import (
    BaseModelOutput,
    ModelOutput,