Skip to content

Commit

Permalink
style
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Feb 4, 2025
1 parent cd3cbb9 commit 2493abe
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/accelerate/state.py
Original file line number Diff line number Diff line change
Expand Up @@ -297,7 +297,7 @@ def __init__(self, cpu: bool = False, **kwargs):

if self.device.type == "hpu":
# we should do this in optimum-habana somehow and not here
from optimum.habana.distributed import parallel_state
from optimum.habana.distributed import parallel_state # noqa: F401

if self.distributed_type != DistributedType.DEEPSPEED:
context_parallel_size = 1
Expand Down

0 comments on commit 2493abe

Please sign in to comment.