Skip to content

GitLab

Unverified Commit bfb1895e authored by Omar Sanseviero's avatar Omar Sanseviero Committed by GitHub
Browse files

Import deepspeed utilities from integrations (#25919) 

Follow up from #25599
parent eb984418
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/vits/modeling_vits.py
View file @ bfb1895e
...@@ -24,7 +24,7 @@ import torch.utils.checkpoint ...@@ -24,7 +24,7 @@ import torch.utils.checkpoint
from torch import nn from torch import nn
from ...activations import ACT2FN from ...activations import ACT2FN
from ...deepspeed import is_deepspeed_zero3_enabled from ...integrations.deepspeed import is_deepspeed_zero3_enabled
from ...modeling_outputs import ( from ...modeling_outputs import (
BaseModelOutput, BaseModelOutput,
ModelOutput, ModelOutput,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment