Export error for SANA

[alexemme]

Hello,
I'm trying to export a diffusion model to ONNX format for the first time.
The model is SANA: https://github.com/NVlabs/Sana

I tried running the following command:

optimum-cli export onnx --model ".../weights/Sana_600M_512px_diffusers" --task default onnx_output/

However, I get this error:

AttributeError: 'Gemma2Config' object has no attribute 'projection_dim'

It seems the model isn't supported.
Does anyone have any advice on how to proceed?

[IlyasMoutawwakil]

Hi ! I moved your issue to the optimum-onnx repo
The Sana model was never added in optimum-onnx but it should be simple to add following the openvino exporter integration.