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

Swin2sr onnx #1492

Merged
merged 5 commits into from
Nov 9, 2023
Merged

Swin2sr onnx #1492

merged 5 commits into from
Nov 9, 2023

Conversation

baskrahmer
Copy link
Contributor

Fixes #1317

@baskrahmer baskrahmer marked this pull request as ready for review November 2, 2023 08:49
@xenova
Copy link
Contributor

xenova commented Nov 7, 2023

Thanks so much for this PR @baskrahmer! Testing it out with Transformers.js now 🤗

btw, have you already exported/uploaded any non-random swin2sr models from the hub (e.g., caidas/swin2SR-classical-sr-x2-64)?

I've uploaded transformers.js-compatible checkpoints here

Copy link
Contributor

@xenova xenova left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can confirm the outputs are correct, and I have integrated it into transformers.js already :)

cc @fxmarty for failing test cases (I doubt this PR has anything to do with it though)

Copy link
Contributor

@fxmarty fxmarty left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM thank you for the addition!

optimum/exporters/tasks.py Outdated Show resolved Hide resolved
optimum/exporters/tasks.py Outdated Show resolved Hide resolved
@fxmarty fxmarty merged commit 41347fc into huggingface:main Nov 9, 2023
67 of 68 checks passed
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

Successfully merging this pull request may close these issues.

[Feature request] Image-to-Image for super-resolution
3 participants