lzy-tony
feat: add english ver
d3653d5
raw
history blame
185 Bytes
UNET_CKPT_NAME = "unet_lora.pt"
INSERTED_ATTN_CKPT_NAME = "unet_inserted_attn.pt"
BYT5_CKPT_NAME = "byt5_model.pt"
BYT5_MAPPER_CKPT_NAME = "byt5_mapper.pt"
huggingface_cache_dir = None