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

add upscale image support to AI pipelines #96

Merged
merged 6 commits into from
Jun 10, 2024

Conversation

mikezupper
Copy link
Contributor

add support to upscale images using timbrooks/instruct-pix2pix model

@mikezupper mikezupper requested a review from rickstaa as a code owner June 5, 2024 23:33
Copy link
Collaborator

@eliteprox eliteprox left a comment

Choose a reason for hiding this comment

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

These changes look good, tested successfully. There is just one change needed to add the default containerHostPort for the new upscale pipeline.

I opened a PR to your branch to add it, please review mikezupper#4

Copy link
Collaborator

@eliteprox eliteprox left a comment

Choose a reason for hiding this comment

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

LGTM! Thanks for the PR @mikezupper

@rickstaa rickstaa force-pushed the feature/add-image-upscale branch from b939673 to 1c5da15 Compare June 10, 2024 12:02
@rickstaa
Copy link
Member

@mikezupper I made some small changes and updated the code with the latest version but apart from that this pull request seems good for a merge 👍🏻. Thanks!

This commit fixes some small bugs and improves the code formatting so
that it is more in line with the other pipelines.
@rickstaa rickstaa force-pushed the feature/add-image-upscale branch from f7c5139 to ae4f3b3 Compare June 10, 2024 12:44
This commit ensures that the `gen_openapi.py` file also creates the
OpenAPI spec for the upscaling route. It also updates the Golang client
bindings.
@rickstaa
Copy link
Member

rickstaa commented Jun 10, 2024

@mikezupper, I noticed that this pull request did not work for me with go-livepeer. It seems you may have forgotten to add the route to the gen_openapi.py script or manually update the specs. I fixed the issues and now it works correctly. 👍🏻

I'm very happy with the cleanliness of the implementation. This serves as an excellent example of a pull request for adding a new pipeline. 🎉

This commit ensures that the right client request and response types
are used.
@rickstaa rickstaa merged commit 45687a7 into livepeer:main Jun 10, 2024
1 check passed
eliteprox added a commit to eliteprox/ai-worker that referenced this pull request Jun 10, 2024
* add upscale image support using stabilityai/stable-diffusion-x4-upscaler model

* Add host port for upscale pipeline

* added model download to the dl_checkpoints.sh

* fix: fix some small bugs and improve formatting

This commit fixes some small bugs and improves the code formatting so
that it is more in line with the other pipelines.

* fix: ensure upscaling OpenAPI spec gets created

This commit ensures that the `gen_openapi.py` file also creates the
OpenAPI spec for the upscaling route. It also updates the Golang client
bindings.

* fix(worker): fix incorrect automatic client types

This commit ensures that the right client request and response types
are used.

---------

Co-authored-by: Elite Encoder <john@eliteencoder.net>
Co-authored-by: Rick Staa <rick.staa@outlook.com>
eliteprox added a commit to eliteprox/ai-worker that referenced this pull request Jul 26, 2024
* add upscale image support using stabilityai/stable-diffusion-x4-upscaler model

* Add host port for upscale pipeline

* added model download to the dl_checkpoints.sh

* fix: fix some small bugs and improve formatting

This commit fixes some small bugs and improves the code formatting so
that it is more in line with the other pipelines.

* fix: ensure upscaling OpenAPI spec gets created

This commit ensures that the `gen_openapi.py` file also creates the
OpenAPI spec for the upscaling route. It also updates the Golang client
bindings.

* fix(worker): fix incorrect automatic client types

This commit ensures that the right client request and response types
are used.

---------

Co-authored-by: Elite Encoder <john@eliteencoder.net>
Co-authored-by: Rick Staa <rick.staa@outlook.com>
@mikezupper mikezupper deleted the feature/add-image-upscale branch August 14, 2024 13:12
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.

3 participants