Some comments to say what the vram state options mean.

This commit is contained in:
comfyanonymous 2023-06-04 17:51:04 -04:00
parent 126b4050dc
commit fed0a4dd29
1 changed files with 3 additions and 3 deletions

View File

@ -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