Update app.py

app.py

@@ -107,11 +107,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