298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-19 10:53:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Some comments to say what the vram state options mean.

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/model_management.py
View File

@ -4,12 +4,12 @@ from comfy.cli_args import args
import torch import torch
class VRAMState(Enum): class VRAMState(Enum):
DISABLED = 0 DISABLED = 0 #No vram present: no need to move models to vram
NO_VRAM = 1 NO_VRAM = 1 #Very low vram: enable all the options to save vram
LOW_VRAM = 2 LOW_VRAM = 2
NORMAL_VRAM = 3 NORMAL_VRAM = 3
HIGH_VRAM = 4 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): class CPUState(Enum):
GPU = 0 GPU = 0