RWKV-Runner/backend-python/global_var.py
2023-05-07 17:27:54 +08:00

28 lines
403 B
Python

from enum import Enum, auto
Model = 'model'
Model_Status = 'model_status'
class ModelStatus(Enum):
Offline = auto()
Loading = auto()
Working = auto()
def init():
global GLOBALS
GLOBALS = {}
set(Model_Status, ModelStatus.Offline)
def set(key, value):
GLOBALS[key] = value
def get(key):
if key in GLOBALS:
return GLOBALS[key]
else:
return None