Update app.py

app.py

@@ -132,20 +132,17 @@ eulera_scheduler = EulerAncestralDiscreteScheduler.from_pretrained("stabilityai/
 # Download the model files
 ckpt_dir_pony = snapshot_download(repo_id="John6666/pony-realism-v21main-sdxl")
 ckpt_dir_cyber = snapshot_download(repo_id="John6666/cyberrealistic-pony-v61-sdxl")
-ckpt_dir_stallion = snapshot_download(repo_id="John6666/stallion-dreams-pony-realistic-v1-sdxl")
 ckpt_dir_stable = snapshot_download(repo_id="stabilityai/stable-diffusion-xl-base-1.0")
 
 
 # Load the models
 vae_pony = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_pony, "vae"), torch_dtype=torch.float16)
 vae_cyber = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_cyber, "vae"), torch_dtype=torch.float16)
-vae_stallion = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_stallion, "vae"), torch_dtype=torch.float16)
 vae_stable = AutoencoderKL.from_pretrained(os.path.join(ckpt_dir_stable, "vae"), torch_dtype=torch.float16)
 
 
 controlnet_pony = ControlNetModel.from_pretrained("xinsir/controlnet-union-sdxl-1.0", torch_dtype=torch.float16)
 controlnet_cyber = ControlNetModel.from_pretrained("xinsir/controlnet-union-sdxl-1.0", torch_dtype=torch.float16)
-controlnet_stallion = ControlNetModel.from_pretrained("xinsir/controlnet-union-sdxl-1.0", torch_dtype=torch.float16)
 controlnet_stable = ControlNetModel.from_pretrained("xinsir/controlnet-union-sdxl-1.0", torch_dtype=torch.float16)
 
 pipe_pony = StableDiffusionXLControlNetPipeline.from_pretrained(
@@ -154,9 +151,6 @@ pipe_pony = StableDiffusionXLControlNetPipeline.from_pretrained(
 pipe_cyber = StableDiffusionXLControlNetPipeline.from_pretrained(
     ckpt_dir_cyber, controlnet=controlnet_cyber, vae=vae_cyber, torch_dtype=torch.float16, scheduler=eulera_scheduler
 )
-pipe_stallion = StableDiffusionXLControlNetPipeline.from_pretrained(
-    ckpt_dir_stallion, controlnet=controlnet_stallion, vae=vae_stallion, torch_dtype=torch.float16, scheduler=eulera_scheduler
-)
 pipe_stable = StableDiffusionXLControlNetPipeline.from_pretrained(
     ckpt_dir_stable, controlnet=controlnet_stable, vae=vae_stable, torch_dtype=torch.float16, scheduler=eulera_scheduler
 )
@@ -248,8 +242,6 @@ def run(
         pipe = pipe_pony
     elif model_choice == "Cyber Realistic Pony v61":
         pipe = pipe_cyber
-    elif model_choice == "Stallion Dreams Pony Realistic v1":
-        pipe = pipe_stallion
     else:
         raise ValueError("Invalid model choice")