Update app.py

app.py

@@ -106,11 +106,11 @@ 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)
+# unet = UNet2DConditionModel.from_pretrained(unet_path, torch_dtype=torch.float16)
 
 pipe = StableDiffusionXLPipeline.from_pretrained(
     base_model_path,
-    unet=unet,
+    # unet=unet,
     scheduler=scheduler,
     torch_dtype=torch.float16, 
     use_safetensors=True