From 3d3dc8a076b42cbd4c449d62ec04f77b830f3cc4 Mon Sep 17 00:00:00 2001 From: wailovet Date: Thu, 15 Aug 2024 01:44:43 +0800 Subject: [PATCH] 'update' --- mz_fluxext_core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mz_fluxext_core.py b/mz_fluxext_core.py index c17d3fa..550d989 100644 --- a/mz_fluxext_core.py +++ b/mz_fluxext_core.py @@ -39,7 +39,7 @@ def Flux1PartialLoad_Patch(args={}): model.model.diffusion_model.double_blocks[layer_start:layer_start + layer_size].to("cpu") torch.cuda.empty_cache() - gc.collect() + # gc.collect() def double_blocks_to_cuda(layer_start=0, layer_size=-1): if layer_size == -1: @@ -55,7 +55,7 @@ def Flux1PartialLoad_Patch(args={}): model.model.diffusion_model.single_blocks[layer_start:layer_start + layer_size].to("cpu") torch.cuda.empty_cache() - gc.collect() + # gc.collect() def single_blocks_to_cuda(layer_start=0, layer_size=-1): if layer_size == -1: