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

feat: optimize model management of nsfw image censoring #2960

Merged

Conversation

mashb1t
Copy link
Collaborator

@mashb1t mashb1t commented May 19, 2024

NSFW censoring now follows general Fooocus model management principles (improved loading and offloading, GPU now handles the processing if possible.)
Includes code optimisations for reusability.

1 shot values, no avg, tested with RTX 3080 and DDR4 RAM:

  • only RAM 240.42 seconds (before adjustments)
  • only GPU 216.37 seconds
  • moving model between RAM / GPU 215.90 seconds (basically same as GPU only)

improved loading and offloading, GPU now handles the processing if possible
@mashb1t mashb1t requested a review from lllyasviel as a code owner May 19, 2024 16:01
@mashb1t mashb1t changed the title feat: optimize nsfw image censoring feat: optimize model management of nsfw image censoring May 19, 2024
@mashb1t mashb1t merged commit f31bec9 into lllyasviel:develop May 19, 2024
@mashb1t mashb1t added this to the 2.4.0 milestone May 19, 2024
mashb1t added a commit that referenced this pull request May 19, 2024
now follows general Fooocus model management principles + includes code optimisations for reusability
csokun pushed a commit to csokun/Fooocus that referenced this pull request Jun 16, 2024
now follows general Fooocus model management principles + includes code optimisations for reusability
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