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