extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-06-05 01:17:18 +02:00
Code Issues 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 user_stopped = True
if context.partial_x_samples is not None: if context.partial_x_samples is not None:
images = latent_samples_to_images(context, context.partial_x_samples) images = latent_samples_to_images(context, context.partial_x_samples)
context.partial_x_samples = None
finally: 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 return images, user_stopped