Import deepspeed utilities from integrations (huggingface#25919)

Follow up from huggingface#25599
blbadger · Nov 8, 2023 · 5c19e17 · 5c19e17
1 parent 4eff608
commit 5c19e17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/models/vits/modeling_vits.py b/src/transformers/models/vits/modeling_vits.py
@@ -24,7 +24,7 @@
 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,