(ace_step) C:\AI\ACE-Step>python trainer.py --dataset_path "C:\Users\DjZaN\OneDrive\Desktop\Zen\dataset.json" --checkpoint_dir "C:\Users\DjZaN\.cache\ace-step\checkpoints" --exp_name "DaltOniX"
C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\requests\__init__.py:86: RequestsDependencyWarning: Unable to find acceptable character detection dependency (chardet or charset_normalizer).
  warnings.warn(
[2025-05-09 16:14:39,198] [INFO] [real_accelerator.py:158:get_accelerator] Setting ds_accelerator to cuda (auto detect)
W0509 16:14:39.808000 11976 site-packages\torch\distributed\elastic\multiprocessing\redirects.py:29] NOTE: Redirects are currently not supported in Windows or MacOs.
Traceback (most recent call last):
  File "C:\AI\ACE-Step\trainer.py", line 861, in <module>
    main(args)
  File "C:\AI\ACE-Step\trainer.py", line 813, in main
    trainer = Trainer(
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\pytorch_lightning\utilities\argparse.py", line 70, in insert_env_defaults
    return fn(self, **kwargs)
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\pytorch_lightning\trainer\trainer.py", line 404, in __init__
    self._accelerator_connector = _AcceleratorConnector(
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\pytorch_lightning\trainer\connectors\accelerator_connector.py", line 157, in __init__
    self._init_strategy()
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\pytorch_lightning\trainer\connectors\accelerator_connector.py", line 480, in _init_strategy
    self.strategy = StrategyRegistry.get(self._strategy_flag)
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\lightning_fabric\strategies\registry.py", line 94, in get
    return data["strategy"](**data["init_params"])
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\pytorch_lightning\strategies\deepspeed.py", line 307, in __init__
    import deepspeed
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\__init__.py", line 22, in <module>
    from . import module_inject
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\module_inject\__init__.py", line 6, in <module>
    from .replace_module import replace_transformer_layer, revert_transformer_layer, ReplaceWithTensorSlicing, GroupQuantizer, generic_injection
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\module_inject\replace_module.py", line 587, in <module>
    from ..pipe import PipelineModule
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\pipe\__init__.py", line 6, in <module>
    from ..runtime.pipe import PipelineModule, LayerSpec, TiedLayerSpec
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\runtime\pipe\__init__.py", line 6, in <module>
    from .module import PipelineModule, LayerSpec, TiedLayerSpec
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\runtime\pipe\module.py", line 19, in <module>
    from ..activation_checkpointing import checkpointing
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\runtime\activation_checkpointing\checkpointing.py", line 26, in <module>
    from deepspeed.runtime.config import DeepSpeedConfig
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\runtime\config.py", line 41, in <module>
    from ..elasticity import (
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\elasticity\__init__.py", line 10, in <module>
    from .elastic_agent import DSElasticAgent
  File "C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\deepspeed\elasticity\elastic_agent.py", line 9, in <module>
    from torch.distributed.elastic.agent.server.api import log, _get_socket_with_port
ImportError: cannot import name 'log' from 'torch.distributed.elastic.agent.server.api' (C:\Users\DjZaN\miniconda3\envs\ace_step\lib\site-packages\torch\distributed\elastic\agent\server\api.py)