Fix #6

2025-06-12 21:22:11 -07:00
parent 90f6043f46
commit ac5ffb4ed0
2 changed files with 3 additions and 5 deletions
@@ -51,8 +51,7 @@ class UnloadModelNode:
            torch.cuda.ipc_collect()
        except:
            print("   - Unable to clear cache")
-        #time.sleep(2) # why?
-        return (list(kwargs.values()))
+        return (kwargs.get("value"),)

 class UnloadAllModelsNode:
    @classmethod
@@ -81,8 +80,7 @@ class UnloadAllModelsNode:
            torch.cuda.ipc_collect()
        except:
            print("   - Unable to clear cache")
-        #time.sleep(2) # why?
-        return (list(kwargs.values()))
+        return (kwargs.get("value"),)


 NODE_CLASS_MAPPINGS = {