From a5e413b575e74e02e20b191283cb163cc6b0ff95 Mon Sep 17 00:00:00 2001
From: GiusTex <112352961+GiusTex@users.noreply.github.com>
Date: Fri, 18 Oct 2024 09:05:41 +0200
Subject: [PATCH] fixed cuda not available on device

---
 nodes.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/nodes.py b/nodes.py
index c1bd7b8..3c33382 100644
--- a/nodes.py
+++ b/nodes.py
@@ -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},)