Fix import.

This commit is contained in:
comfyanonymous 2023-05-05 00:19:35 -04:00
parent 1a31020081
commit 6ee11d7bc0

View File

@ -1,6 +1,6 @@
import psutil
from enum import Enum
from .cli_args import args
from comfy.cli_args import args
class VRAMState(Enum):
CPU = 0