Files
FuouM-FM_nodes/constants.py
T
2024-08-03 22:52:38 +07:00

11 lines
414 B
Python

WFEN_MODEL_PATH = "models/wfen/WFEN.pth"
REALVIFORMER_MODEL_PATH = "models/realviformer/weights.pth"
PROPIH_VGG_MODEL_PATH = "models/propih/vgg_normalised.pth"
PROPIH_G_MODEL_PATH = "models/propih/latest_net_G.pth"
VFI_MAMBA_MODELS = ["VFIMamba", "VFIMamba_S"]
VFI_MAMBA_PATHS = {
"VFIMamba": "models/vfimamba/VFIMamba.pkl",
"VFIMamba_S": "models/vfimamba/VFIMamba_S.pkl",
}
VFI_MAMBA_DEFAULT = "VFIMamba"