Running into this issue. Any ideas for a clean resolve?
File "/opt/conda/lib/python3.9/site-packages/accelerate/accelerator.py", line 33, in <module>
from .checkpointing import load_accelerator_state, load_custom_state, save_accelerator_state, save_custom_state
File "/opt/conda/lib/python3.9/site-packages/accelerate/checkpointing.py", line 24, in <module>
from .utils import (
File "/opt/conda/lib/python3.9/site-packages/accelerate/utils/__init__.py", line 109, in <module>
from .launch import (
File "/opt/conda/lib/python3.9/site-packages/accelerate/utils/launch.py", line 27, in <module>
from ..utils.other import merge_dicts
File "/opt/conda/lib/python3.9/site-packages/accelerate/utils/other.py", line 28, in <module>
from deepspeed import DeepSpeedEngine
File "/opt/conda/lib/python3.9/site-packages/deepspeed/__init__.py", line 16, in <module>
from .runtime.engine import DeepSpeedEngine, DeepSpeedOptimizerCallable, DeepSpeedSchedulerCallable
File "/opt/conda/lib/python3.9/site-packages/deepspeed/runtime/engine.py", line 25, in <module>
from deepspeed.runtime.utils import see_memory_usage, get_ma_status, DummyOptim
File "/opt/conda/lib/python3.9/site-packages/deepspeed/runtime/utils.py", line 18, in <module>
from torch._six import inf
ModuleNotFoundError: No module named 'torch._six'
Running into this issue. Any ideas for a clean resolve?