Skip to content

Commit

Permalink
fix: Support different versions of accelerate
Browse files Browse the repository at this point in the history
  • Loading branch information
rrutmann committed Jun 5, 2024
1 parent d2f6dd9 commit 3f5ffab
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/modalities/running_env/fsdp/fsdp_auto_wrapper.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from typing import Callable, List

import torch.nn as nn
from accelerate.utils.dataclasses import get_module_class_from_name
from accelerate import FullyShardedDataParallelPlugin
from torch.distributed.fsdp.wrap import transformer_auto_wrap_policy

from modalities.config.lookup_enum import LookupEnum
Expand Down

0 comments on commit 3f5ffab

Please sign in to comment.