search

Vchitect / Vchitect-2.0

Vchitect-2.0: Parallel Transformer for Scaling Up Video Diffusion Models
https://vchitect.intern-ai.org.cn/
Apache License 2.0
534 stars 13 forks source link

Runtime error: cannot import name 'PrepareModuleOutput' from 'torch.distributed.tensor.parallel' #5

Closed yuanzhi-zhu closed 3 days ago

yuanzhi-zhu commented 5 days ago

Thank your for opening source such a great work

I got the above error when running the inference code.

I installed the env followed the instructions in Readme, and it seems that we need PyTorch > 2.1 to have PrepareModuleOutput in torch.distributed.tensor.parallel.

Did I missed something here?

WeizhenEricFang commented 3 days ago

Have you got any solution? I got the same error

yuanzhi-zhu commented 3 days ago

@WeizhenEricFang Just install PyTorch > 2.1 and resolve some other conflicts