huggingface / optimum-neuron

Easy, fast and very cheap training and inference on AWS Trainium and Inferentia chips.

Apache License 2.0

176 stars 51 forks source link

enable unequal height and width #592

Closed yahavb closed 1 month ago

yahavb commented 1 month ago

What does this PR do?

enable unequal height and width, currently it is blocked and neuron is going to add that support in the next SDK version.

Fixes # (issue)

Before submitting

[x ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
[ x] Did you make sure to update the documentation with your changes?
[ x] Did you write any new necessary tests?

HuggingFaceDocBuilderDev commented 1 month ago

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

yahavb commented 1 month ago

Anything else is needed to merge this one?

JingyaHuang commented 1 month ago

Thanks @yahavb for the PR, merged!