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

Studio2/SD/API: Exclude more static_kwargs in StableDiffusion.init for now #2096

Open
wants to merge 1 commit into
base: sd-studio2
Choose a base branch
from

Conversation

one-lithe-rune
Copy link
Contributor

Motivation

Although the changes in #2091 got things to failing at the point of iree-compile of unet for me, the merged version considers vae_encode before unet. It turns out vae_encode has a similar problem with including a precision argument when main branch turbine_models.custom_models.sd_inference doesn't yet include it.

This comments out precision for all the blocks in the the static_kwargs variable which defines how the the various pipeline models are called, and gets us back to an iree-compile failure again.

Changes

  • Comment out precision being set in static_kwargs for vae_encode when setting up a StableDiffusion pipeline
  • Comment out precision being set in static_kwargs for vae_decode when setting up a StableDiffusion pipeline

Possible Problems/Concerns

  • This is will need to be reverted once support for precision is landed in the Turbine models, I've added a TODO comment saying that.

- Comment out precision being set in static_kwargs for vae_encode
when setting up a StableDiffusion pipeline
- Comment out precision being set in static_kwargs for vae_decode
when setting up a StableDiffusion pipeline
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.

1 participant