ModuleNotFoundError: No module named 'torch.distributed.tensor.parallel._utils'

[sansure]

Building a conda environment for diffsynth-engine with torch=2.9.1+cu128, but encountering an issue: ModuleNotFoundError: No module named 'torch.distributed.tensor.parallel._utils'. May I ask which part of the environment is mismatched, and what is the optimal environment configuration? Thank you！

[sansure]

Use command：
conda create -n diffsynth_engine python=3.10 -y
pip install diffsynth-engine==0.4.1

[akaitsuki-ii]

@sansure The module could not be found due to API changed in torch 2.9, please use torch<2.9 instead, will fix later