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.

Sign up for GitHub

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

Change NSFW Model #307

Merged
merged 3 commits into from
Oct 16, 2024
Merged

Change NSFW Model #307

merged 3 commits into from
Oct 16, 2024

Conversation

ryantwolf
Copy link
Collaborator

@ryantwolf ryantwolf commented Oct 16, 2024 •
edited
Loading

Description

Changes the NSFW model from one that takes ViT-H embeddings as input into one that takes ViT-L embeddings as input.

Usage

No change

Checklist

  • I am familiar with the Contributing Guide.
  • New or Existing tests cover these changes.
  • The documentation is up to date with these changes.

@ryantwolf
Change download for NSFW model
8df7f71 
Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
@ryantwolf
Fix model init
4100d51 
Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
@ryantwolf
Fix embedding size
5d740ba 
Signed-off-by: Ryan Wolf <rywolf@nvidia.com>
VibhuJawa
VibhuJawa approved these changes Oct 16, 2024
View reviewed changes
Copy link
Collaborator

@VibhuJawa VibhuJawa 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 fixing this

@ryantwolf ryantwolf merged commit 37058a9 into r0.5.0 Oct 16, 2024
3 checks passed
@ryantwolf ryantwolf deleted the rywolf/nsfw-model branch October 16, 2024 22:41
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@VibhuJawa VibhuJawa VibhuJawa approved these changes

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@ryantwolf @VibhuJawa