-
Notifications
You must be signed in to change notification settings - Fork 6.1k
Update pipeline_flax_stable_diffusion_controlnet.py #3306
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
Conversation
The documentation is not available anymore as the PR was closed or merged. |
@yiyixuxu can you check? |
Gentle ping @yiyixuxu - we can also just leave this one stale for now and ask for help from the community if you prefer |
It was just a quick fix for the jax-diffusers hackathon. |
@jfozard I think it's good. can you resolve the conflicts and make sure all the tests pass? |
a753cc5
to
34d14d7
Compare
Change type of images array from jax.numpy.array to numpy.ndarray to permit in-place modification of the array when the safety checker detects a NSFW image.
Sorry - I reverted the commit in my branch using the github interface, and didn't realize this would close the PR. |
Update pipeline_flax_controlnet.py Change type of images array from jax.numpy.array to numpy.ndarray to permit in-place modification of the array when the safety checker detects a NSFW image.
Update pipeline_flax_controlnet.py Change type of images array from jax.numpy.array to numpy.ndarray to permit in-place modification of the array when the safety checker detects a NSFW image.
np.asarray makes a read-only copy of a jax array. This then breaks at line 473 if a nsfw image is detected