From d001d4ae064f550f3ff4b4fc407c9755a29215f0 Mon Sep 17 00:00:00 2001
From: Sumeth Sathnindu <65599800+LucipherDev@users.noreply.github.com>
Date: Thu, 16 Jan 2025 00:50:03 +0530
Subject: [PATCH] Bug fix

---
 nodes.py | 7 -------
 1 file changed, 7 deletions(-)

diff --git a/nodes.py b/nodes.py
index bdd6e63..c2ea1de 100644
--- a/nodes.py
+++ b/nodes.py
@@ -264,13 +264,6 @@ class TangoFluxVAEDecodeAndPlay:
 
         try:
             with torch.no_grad():
-                try:
-                    print("Inducing OOM for testing...")
-                    oom_tensor = torch.empty((1610612736,), dtype=torch.float32, device="cuda")
-                except torch.OutOfMemoryError:
-                    print("OOM successfully induced for testing.")
-                    raise torch.OutOfMemoryError
-                
                 for latent in latents:
                     torch.cuda.empty_cache()