Pytorch version problem

[MrDasix]

When running

examples/wan_text_to_video.py

with pytorch 2.9.0

i get the error:

  File "/app/diffsynth-engine/diffsynth_engine/utils/parallel.py", line 14, in <module>
    from torch.distributed.tensor.parallel._utils import _validate_tp_mesh_dim
ModuleNotFoundError: No module named 'torch.distributed.tensor.parallel._utils'

[akaitsuki-ii]

@MrDasix I notice that PyTorch 2.9 have some API changed, please switch to older version, will fix the issue later

[MrDasix]

Thanks! Now it does not give this error. Now the error given is that no module "vsa" is installed (not mentioned in the requirements) and when i run the same code after pip install vsa it gives a circular import error

[akaitsuki-ii]

@MrDasix It's an known issue and I am just working to fix it. This branch should work well and we are going to merge it to main branch tomorrow.

[yq-xfl]

@MrDasix I notice that PyTorch 2.9 have some API changed, please switch to older version, will fix the issue later

这个修复了吗

[akaitsuki-ii]

@MrDasix 还没时间去修这个API问题，推荐先使用torch<=2.8