From 5c19e175345e18837beea33cc3e567f2130c9369 Mon Sep 17 00:00:00 2001 From: Omar Sanseviero Date: Mon, 4 Sep 2023 15:03:48 +0200 Subject: [PATCH] Import deepspeed utilities from integrations (#25919) 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 7f3e90194246bf..4b5dc9a132df7a 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,