From fed0a4dd29852e4808382ef9428a2256214667bf Mon Sep 17 00:00:00 2001
From: comfyanonymous <comfyanonymous@protonmail.com>
Date: Sun, 4 Jun 2023 17:51:04 -0400
Subject: [PATCH] Some comments to say what the vram state options mean.

---
 comfy/model_management.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/comfy/model_management.py b/comfy/model_management.py
index a492ca6b..1a8a1be1 100644
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -4,12 +4,12 @@ from comfy.cli_args import args
 import torch
 
 class VRAMState(Enum):
-    DISABLED = 0
-    NO_VRAM = 1
+    DISABLED = 0    #No vram present: no need to move models to vram
+    NO_VRAM = 1     #Very low vram: enable all the options to save vram
     LOW_VRAM = 2
     NORMAL_VRAM = 3
     HIGH_VRAM = 4
-    SHARED = 5
+    SHARED = 5      #No dedicated vram: memory shared between CPU and GPU but models still need to be moved between both.
 
 class CPUState(Enum):
     GPU = 0