From e485f7b242261705b3d977af97dad5ee66da00c6 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 19 Sep 2023 12:24:07 -0700 Subject: [PATCH] i (#438) --- fooocus_version.py | 2 +- modules/default_pipeline.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/fooocus_version.py b/fooocus_version.py index e6e36dd5..c7e2a7da 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.0.66' +version = '2.0.67' diff --git a/modules/default_pipeline.py b/modules/default_pipeline.py index f19e9f76..70d75e10 100644 --- a/modules/default_pipeline.py +++ b/modules/default_pipeline.py @@ -3,6 +3,7 @@ import os import torch import modules.path import modules.virtual_memory as virtual_memory +import comfy.model_management from comfy.model_base import SDXL, SDXLRefiner from modules.patch import cfg_patched, patched_model_function @@ -262,4 +263,6 @@ def process_diffusion(positive_cond, negative_cond, steps, switch, width, height decoded_latent = core.decode_vae(vae=xl_base_patched.vae, latent_image=sampled_latent, tiled=tiled) images = core.pytorch_to_numpy(decoded_latent) + + comfy.model_management.soft_empty_cache() return images