EX4L commited on
Commit
abe1951
·
verified ·
1 Parent(s): c23c178

Trying out new models

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ download_file("https://huggingface.co/ai-forever/Real-ESRGAN/resolve/main/RealES
50
  # Download the model files
51
  ckpt_dir_pony = snapshot_download(repo_id="John6666/pony-realism-v21main-sdxl")
52
  ckpt_dir_cyber = snapshot_download(repo_id="John6666/cyberrealistic-pony-v63-sdxl")
53
- ckpt_dir_stallion = snapshot_download(repo_id="John6666/stallion-dreams-pony-realistic-v1-sdxl")
54
 
55
  # Load the models
56
  vae_pony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_pony, "vae"), torch_dtype=torch.float16)
 
50
  # Download the model files
51
  ckpt_dir_pony = snapshot_download(repo_id="John6666/pony-realism-v21main-sdxl")
52
  ckpt_dir_cyber = snapshot_download(repo_id="John6666/cyberrealistic-pony-v63-sdxl")
53
+ ckpt_dir_stallion = snapshot_download(repo_id="frosting-ai/compassmix-xl-lightning")
54
 
55
  # Load the models
56
  vae_pony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_pony, "vae"), torch_dtype=torch.float16)