Update app.py

app.py

@@ -105,21 +105,20 @@ base_model_path: str = "stabilityai/stable-diffusion-xl-base-1.0"
 refiner_model_path: str = "stabilityai/stable-diffusion-xl-refiner-1.0" 
 unet_path: str = "cosmicman/CosmicMan-SDXL"
 SCHEDULER = schedule_map[schedule]
-scheduler = SCHEDULER.from_pretrained(base_model_path, subfolder="scheduler", torch_dtype=torch.float16)
-# unet = UNet2DConditionModel.from_pretrained(unet_path, torch_dtype=torch.float16)
+scheduler = SCHEDULER.from_pretrained(base_model_path, subfolder="scheduler")
+# unet = UNet2DConditionModel.from_pretrained(unet_path)
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
     base_model_path,
     # unet=unet,
     scheduler=scheduler,
-    torch_dtype=torch.float16, 
     use_safetensors=True
 ).to("cuda")
 pipe.watermark = NoWatermark()
 refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
     base_model_path, # we found use base_model_path instead of refiner_model_path may get a better performance
     scheduler=scheduler,
-    torch_dtype=torch.float16, use_safetensors=True
+    use_safetensors=True
 ).to("cuda")
 refiner.watermark = NoWatermark()
 print("Finish Loading Model!")