search

Tencent / HunyuanDiT

Hunyuan-DiT : A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understanding
https://dit.hunyuan.tencent.com/
Other
3.32k stars 285 forks source link

torch version issue #119

Closed sahilqure closed 2 months ago

sahilqure commented 3 months ago

ModuleNotFoundError: No module named 'torch._six' Using CUDA 12.1 Can u update the repo for latest CUDA

zml-ai commented 2 months ago

Hi, we provide Dockers for CUDA 11/12, which you can use to simplify environment installation. For more details, please see: https://github.com/Tencent/HunyuanDiT?tab=readme-ov-file#installation-guide-for-linux