From f221a2cc1feb2f3f58da1d0540a4bf1134190984 Mon Sep 17 00:00:00 2001 From: Omar Sanseviero Date: Fri, 1 Sep 2023 21:43:16 +0200 Subject: [PATCH] Import deepspeed utilities from integrations Follow up from #25599 --- src/transformers/models/vits/modeling_vits.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/vits/modeling_vits.py b/src/transformers/models/vits/modeling_vits.py index 876abc870470..d7c1f84e1f20 100644 --- 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,