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"