Update app.py

app.py

@@ -29,7 +29,8 @@ print(f"dtype: {dtype}")
 print(f"low memory: {LOW_MEMORY}")
 
 
-model = "stabilityai/stable-diffusion-xl-base-1.0"
+# model = "stabilityai/stable-diffusion-xl-base-1.0"
+model = "cagliostrolab/animagine-xl-4.0"
 # model = "stabilityai/sdxl-turbo"
 # vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=dtype)
 scheduler = DDIMScheduler.from_pretrained(model, subfolder="scheduler")
@@ -46,7 +47,7 @@ pipe = StableDiffusionXLControlNetImg2ImgPipeline.from_pretrained(
     model,
     controlnet=controlnet,
     torch_dtype=dtype,
-    variant="fp16",
+    #variant="fp16",
     use_safetensors=True,
     scheduler=scheduler,
 )