X Tutup
The Wayback Machine - https://web.archive.org/web/20221231211226/https://github.com/CompVis/stable-diffusion/pull/36
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

Jump to bottom

[Safety Checker] Add Safety Checker Module #36

Merged
merged 6 commits into from Aug 22, 2022
Merged

[Safety Checker] Add Safety Checker Module #36

merged 6 commits into from Aug 22, 2022
+28 −2

Conversation

patrickvonplaten
Copy link
Contributor

@patrickvonplaten patrickvonplaten commented Aug 19, 2022 •
edited

With @anton-l and @patil-suraj, we've made sure that the safety_checker works correctly see here: huggingface/diffusers#219

The only thing we're not 100% sure about is whether this line https://github.com/huggingface/diffusers/blob/89e9521048067acacfdcbc2b985af8f6b155cfb6/src/diffusers/pipelines/stable_diffusion/safety_checker.py#L65 that defines the threshold is correct

patrickvonplaten
patrickvonplaten commented Aug 19, 2022
View changes
scripts/txt2img.py Outdated Show resolved Hide resolved
patrickvonplaten
patrickvonplaten commented Aug 19, 2022
View changes
scripts/txt2img.py Outdated Show resolved Hide resolved
patrickvonplaten
patrickvonplaten commented Aug 19, 2022
View changes
scripts/txt2img.py Outdated Show resolved Hide resolved
@rromb rromb self-assigned this Aug 19, 2022
@rromb rromb self-requested a review Aug 19, 2022
rromb
rromb reviewed Aug 19, 2022
View changes
Copy link
Member

@rromb rromb left a comment

Looks good, we could consider making this optional

@patrickvonplaten
Copy link
Contributor Author

patrickvonplaten commented Aug 19, 2022

Happy to adapt the way you prefer

@pesser pesser merged commit a6e2f3b into CompVis:main Aug 22, 2022
@Bendito999
Copy link

Bendito999 commented Aug 22, 2022

Any instructions to adjust this? Emad on twitter had mentioned there was a way to specifically block Clowns for example, which is a fun insight into how this works. Or just a simple 'comment this out to completely take safety off' might also be a good note to have.

enzymezoo-code added a commit to enzymezoo-code/stable-diffusion that referenced this pull request Sep 2, 2022
@enzymezoo-code
Merge pull request CompVis#36 from enzymezoo-code/inpainting_1.0
3ba2f92 
Updating ipynb with colab-convert
@dannydeezy
Copy link

dannydeezy commented Sep 21, 2022

Any instructions to adjust this? Emad on twitter had mentioned there was a way to specifically block Clowns for example, which is a fun insight into how this works. Or just a simple 'comment this out to completely take safety off' might also be a good note to have.

yes, would be nice to be able to remove the safety checker. some of us are adults!

@woctezuma
Copy link

woctezuma commented Sep 21, 2022

Or just a simple 'comment this out to completely take safety off' might also be a good note to have.

yes, would be nice to be able to remove the safety checker. some of us are adults!

You can already remove the safety, it is just that the removal procedure is not explained in the documentation.

@aisensiy
Copy link

aisensiy commented Oct 11, 2022

I saw the model come from https://huggingface.co/CompVis/stable-diffusion-safety-checker/blob/main/pytorch_model.bin and I am wondering where is this model come from. There is some nsfw detector in github but this one looks different (larger).

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@rromb rromb

Assignees

@rromb rromb

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.

None yet
7 participants
@patrickvonplaten @Bendito999 @dannydeezy @woctezuma @aisensiy @rromb @pesser
X Tutup