Update app.py

app.py

@@ -38,18 +38,16 @@ model_tuple = ControlNetModel_Union._load_pretrained_model(
 )
 model = model_tuple[0]  # 取出实际的模型对象
 
-model.to(device="cpu", dtype=torch.float16)
+model.to(device="cpu")
 
 vae = AutoencoderKL.from_pretrained(
-    "madebyollin/sdxl-vae-fp16-fix"
+    "stabilityai/sd-vae-ft-ema"
 ).to("cpu")
 
 pipe = StableDiffusionXLFillPipeline.from_pretrained(
     "SG161222/RealVisXL_V5.0_Lightning",
-    torch_dtype=torch.float16,
     vae=vae,
     controlnet=model,
-    variant="fp16",
 ).to("cpu")
 
 pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)