fixed cuda not available on device

This commit is contained in:
GiusTex
2024-10-18 09:05:41 +02:00
committed by GitHub
parent 491ba204fb
commit a5e413b575
+2 -4
View File
@@ -2,7 +2,7 @@ import torch
import gc
import os
from PIL import Image
from .utils import get_first_folder_list, tensor2pil, pil2tensor, encodeDiffOutpaintPrompt, diffuserOutpaintSamples, get_device_by_name, get_dtype_by_name
from .utils import get_first_folder_list, tensor2pil, pil2tensor, encodeDiffOutpaintPrompt, diffuserOutpaintSamples, get_device_by_name, get_dtype_by_name, clearVram
# Get the absolute path of various directories
@@ -256,8 +256,6 @@ class DiffusersImageOutpaint:
device, steps, controlnet_strength, guidance_scale,
keep_model_device)
del diffusers_outpaint_conditioning
gc.collect()
torch.cuda.empty_cache()
torch.cuda.ipc_collect()
clearVram(device)
return ({"samples":last_rgb_latent},)