JadenFK's picture
Init for demo
640a27b
raw
history blame
532 Bytes
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']
)