0.18

free_vram_hack.py

@@ -0,0 +1,56 @@
+import torch
+import gc
+import requests
+import json
+
+class FreeVRAMNode:
+    @classmethod
+    def INPUT_TYPES(s):
+        return {"required": {"image": ("IMAGE",)}}
+    
+    RETURN_TYPES = ("IMAGE",)
+    FUNCTION = "free_vram"
+    CATEGORY = "memory_management"
+
+    def free_vram(self, image):
+        print("Attempting to free VRAM...")
+        
+        # Clear CUDA cache
+        if torch.cuda.is_available():
+            torch.cuda.empty_cache()
+            print("CUDA cache cleared.")
+        
+        # Run garbage collection
+        collected = gc.collect()
+        print(f"Garbage collector: collected {collected} objects.")
+        
+        # Trigger the HTTP request
+        self.trigger_http_request()
+        
+        # Return the input image unchanged
+        return (image,)
+    
+    def trigger_http_request(self):
+        url = "http://localhost:8188/prompt"
+        headers = {"Content-Type": "application/json"}
+        payload = {
+            "prompt": {
+                "3": {
+                    "inputs": {"text": "free VRAM hack"},
+                    "class_type": "Bjornulf_WriteText",
+                    "_meta": {"title": "✒ Write Text"}
+                },
+                "4": {
+                    "inputs": {"text_value": ["3", 0], "text": "free VRAM hack"},
+                    "class_type": "Bjornulf_ShowText",
+                    "_meta": {"title": "👁 Show (Text)"}
+                }
+            }
+        }
+        
+        try:
+            response = requests.post(url, headers=headers, data=json.dumps(payload))
+            response.raise_for_status()
+            print("HTTP request triggered successfully")
+        except requests.exceptions.RequestException as e:
+            print(f"Failed to trigger HTTP request: {e}")