Trying out new models
Browse files
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="
|
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)
|