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

更新train代码之后,把之前的tensorrt推理代码给覆盖了 #88

Closed Jimzhou82sub closed 3 months ago

Jimzhou82sub commented 3 months ago

如题,当前代码不支持tensorRT推理了

zml-ai commented 3 months ago

We have fixed the bug, details can be found at https://github.com/Tencent/HunyuanDiT/blob/main/hydit/inference.py#L18

Jimzhou82sub commented 3 months ago

请问tensorRT后续会支持到int8 优化吗