Update app.py

app.py

@@ -104,10 +104,10 @@ print("Loading Model!")
 schedule: str = "euler_a"
 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"
+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,