• Your Name's avatar
    Add aggressive VRAM cleanup for model switching · 7c150a4d
    Your Name authored
    - Added _aggressive_vram_cleanup method to properly clear VRAM
    - Moves model to CPU before deletion
    - Deletes pipeline, vae, text_encoder, tokenizer explicitly
    - Multiple rounds of gc.collect()
    - Uses torch.cuda.synchronize() before clearing cache
    - Increased delay to 5 seconds after cleanup
    7c150a4d
Name
Last commit
Last update
.vscode Loading commit data...
.gitignore Loading commit data...
LICENSE.md Loading commit data...
README.md Loading commit data...
aaa Loading commit data...
build.sh Loading commit data...
coder Loading commit data...
coderai Loading commit data...
requirements-nvidia.txt Loading commit data...
requirements-vulkan.txt Loading commit data...
requirements.txt Loading commit data...
requirements.txt~ Loading commit data...