import sys
sys.path.insert(0,'stable_diffusion')
from train_esd import train_esd

ckpt_path = "stable_diffusion/models/ldm/sd-v1-4-full-ema.ckpt"
config_path = "stable_diffusion/configs/stable-diffusion/v1-inference.yaml"
diffusers_config_path = "stable_diffusion/config.json"

train_esd("England",
              'xattn',
              3,
              1,
              1000,
              .003,
              config_path,
              ckpt_path, 
              diffusers_config_path,
              ['cuda', 'cuda']
              )