Fix VRAM spike

I am an idiot
This commit is contained in:
kijai
2024-03-01 19:59:00 +02:00
parent 093baa0957
commit c74b8248a7
+2 -3
View File
@@ -152,7 +152,7 @@ class SUPIR_Upscale:
del supir_state_dict, sdxl_state_dict
comfy.model_management.soft_empty_cache()
self.model.to(device).to(dtype)
self.model.to(dtype).to(device)
if use_tiled_vae:
self.tiled_vae_state = True
@@ -169,7 +169,6 @@ class SUPIR_Upscale:
captions_list = []
captions_list.append(captions)
print(captions_list)
use_linear_CFG = cfg_scale_start > 0
use_linear_control_scale = control_scale_start > 0
@@ -184,7 +183,7 @@ class SUPIR_Upscale:
cfg_scale_start=cfg_scale_start, control_scale_start=control_scale_start)
out.append(samples.squeeze(0).cpu())
print("Sampled image ", i, " out of ", B)
print("Sampled image ", i + 1, " out of ", B)
pbar.update(1)
if not keep_model_loaded:
self.model = None