export onnx when classfier guidance

[smilebetterworld]

@LuChengTHU ， Hi， when using classifer guidance which requires grad calculation, there is some errors when exporting model into onnx , because grad is forbidden when model export phase. How to solver this contradiction？

[freeclouds]

@LuChengTHU ， Hi， when using classifer guidance which requires grad calculation, there is some errors when exporting model into onnx , because grad is forbidden when model export phase. How to solver this contradiction？

你好，我在difusion planner模型从pytorch模型转换到onnx模型过程中也遇到了类似的错误，这个问题你解决了吗？