Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

trt部署时,模型导出onnx失败 #184

Open
linuxmi opened this issue Jul 31, 2024 · 2 comments
Open

trt部署时,模型导出onnx失败 #184

linuxmi opened this issue Jul 31, 2024 · 2 comments

Comments

@linuxmi
Copy link

linuxmi commented Jul 31, 2024

Describe the bug
A clear and concise description of what the bug is.
变量style如何处理
企业微信截图_17224141856142
image

Reproduction
按照教程执行,使用hunyuandit提供docker, 硬件A800,driver version 525.85.12
wget https://dit.hunyuan.tencent.com/download/HunyuanDiT/hunyuan_dit_cu11.tar
https://hf-mirror.com/Tencent-Hunyuan/TensorRT-libs/blob/main/README_zh.md

Tasks

No tasks being tracked yet.
@linuxmi linuxmi changed the title trt部署时,模型到处onnx失败 trt部署时,模型导出onnx失败 Jul 31, 2024
@ZiangWu-77
Copy link

@linuxmi Which model did you use then? I met the same issue when using v1.2(needs no style or style_cond) and changed model to v1.1 and another issue came up LOL.

@zymy-chen
Copy link

遇到了相同的问题,当用v1.2的时候修改export_onnx.py 将ims和style改成None,遇到了另一个报错
image

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants