Fix stable diffusion export

[JingyaHuang]

Fix bugs introduced by the update of the optimum main repo.

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint.

[philschmid]

LGTM

[dacorvo]

In addition to the usual errors we have, there seems to be an issue with stable diffusion export tests: 'GPT2NeuronConfig' has no attribute 'get_mandatory_axes_for_task'.

[JingyaHuang]

In addition to the usual errors we have, there seems to be an issue with stable diffusion export tests: 'GPT2NeuronConfig' has no attribute 'get_mandatory_axes_for_task'.

@dacorvo, it's interesting as GPT2NeuronConfig is not supposed to be used during the sd test 🤔.

Could you give me a snippet so that I can reproduce it? Thanks!

[dacorvo]

In addition to the usual errors we have, there seems to be an issue with stable diffusion export tests: 'GPT2NeuronConfig' has no attribute 'get_mandatory_axes_for_task'.

@dacorvo, it's interesting as GPT2NeuronConfig is not supposed to be used during the sd test 🤔.

Could you give me a snippet so that I can reproduce it? Thanks!

Yes, my bad, this error is for the next test. The stable diffusion error is about a string passed as a list index.
This is the log from the CI: https://github.com/huggingface/optimum-neuron/actions/runs/5598435281/jobs/10238146802?pr=144.