Skip to content

An Extension for Automatic1111 Webui that releases the memory each generation and release vram

License

Notifications You must be signed in to change notification settings

hbulixu/sd-webui-release-memory

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 

Repository files navigation

增加功能,每次使用完后卸载掉checkpoint,减少显存占用

SD Webui Memory Release

This is an Extension for the Automatic1111 Webui, that calls torch.cuda.empty_cache() after each generation.

But Why?

  • A few Reddit posts/comments mention that CUDA sometimes has memory issues, and the webui may cause memory leaks. This Extension tried to mediate that by calling torch.cuda.empty_cache() after each generation.

"Features"

  • Turns out postprocess is not called when a generation fails. Thus, I added a button to manually try releasing the memory.

  • Also comes with another button that unloads and reloads the checkpoint for the maximum cleanse.

    • This just calls the same functions that you can already find in the Actions section of the Settings tab
  • In the System section of the Settings tab, you can also turn on debug log or enable additionally refreshing the torch networks.


[1] I'm 99.99% sure that this Extension helps nothing. You most likely do not need this.

[2] Shout out to @kgmkm_mkgm for sharing this Extension with tens of thousands of people.

[3] Seriously, why are people downloading this? Why are people even Star-ing this? What does this actually help?

About

An Extension for Automatic1111 Webui that releases the memory each generation and release vram

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%