Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add workaround low memory crash during image saving

Browse Source 
Sample model moved to cpu before processing samples to image
This commit is contained in:
ArDiouscuros
2022-09-24 10:29:06 +02:00
committed by Lincoln Stein
parent b40bfb5116
commit 485fe67c92
4 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/dream.py
View File

@ -108,6 +108,8 @@ def main():
# preload the model
gen.load_model()
#set additional option
gen.free_gpu_mem = opt.free_gpu_mem
if not infile:
print(