ComfyUI-StyleTransferPlus/constants.py at main · FuouM/ComfyUI-StyleTransferPlus · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
CAST_TYPES = ["CAST", "UCAST"]
CAST_DEFAULT = "CAST"

CAST_NET_AE_PATH = "latest_net_AE.pth"
CAST_NET_DEC_B_PATH = "latest_net_Dec_B.pth"
CAST_VGG_PATH = "models/vgg_normalised.pth"

EFDM_STYLE_TYPES = ["adain", "adamean", "adastd", "efdm", "hm"]
EFDM_DEFAULT = "efdm"
EFDM_PATH = "models/hm_decoder_iter_160000.pth"

MICROAST_CONTENT_ENCODER_PATH = "models/microast/content_encoder_iter_160000.pth.tar"
MICROAST_STYLE_ENCODER_PATH = "models/microast/style_encoder_iter_160000.pth.tar"
MICROAST_DECODER_PATH = "models/microast/decoder_iter_160000.pth.tar"
MICROAST_MODULATOR_PATH = "models/microast/modulator_iter_160000.pth.tar"

UNIST_PATH = "models/unist/UniST_model.pt"
UNIST_DEC_PATH = "models/unist/dec_r41.pth"
UNIST_ENC_PATH = "models/unist/vgg_r41.pth"

VGG_NORM_CONV51_PATH = "models/vgg_normalised_conv5_1.pth"
AESPA_DEC_PATH = "models/aespa/dec_model.pth"
AESPA_TRANSFORMER_PATH = "models/aespa/transformer_model.pth"

TSSAT_DEC_PATH = "models/tssat/decoder_iter_160000.pth"

AESFA_PATH = "models/aesfa/main.pth"