Update app.py

app.py

@@ -29,7 +29,7 @@ ckpt_name = "Hyper-SDXL-1step-Unet.safetensors"
 
 unet = UNet2DConditionModel.from_config(base, subfolder="unet").to("cuda", torch.float16)
 unet.load_state_dict(load_file(hf_hub_download(repo_name, ckpt_name), device="cuda"))
-pipe_hyper = DiffusionPipeline.from_pretrained(base_model_id, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
+pipe_hyper = StableDiffusionXLPipeline.from_pretrained(base_model_id, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
 pipe_hyper.scheduler = LCMScheduler.from_config(pipe_hyper.scheduler.config)
 pipe_hyper.to("cuda")