holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Proactively delete the partial samples from the callbacks

Browse Source
This commit is contained in:
cmdr2 2023-02-01 11:50:50 +05:30
parent 3e7f14af2c
commit 51443741b8
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ui/easydiffusion/renderer.py
View File

@ -72,9 +72,10 @@ def generate_images_internal(req: GenerateImageRequest, task_data: TaskData, dat
user_stopped = True
if context.partial_x_samples is not None:
images = latent_samples_to_images(context, context.partial_x_samples)
context.partial_x_samples = None
finally:
gc(context)
if hasattr(context, 'partial_x_samples') and context.partial_x_samples is not None:
del context.partial_x_samples
context.partial_x_samples = None
return images, user_stopped