trying to add sdxl

app.py

@@ -184,6 +184,19 @@ model.load_state_dict(state_dict, strict=True)
 
 model = model.to(device)
 
+# Load text-to-image model
+print('Loading text-to-image model ...')
+
+pipe = StableDiffusionXLPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.bfloat16)
+pipe.to(device="cuda", dtype=torch.bfloat16)
+
+unet_state = load_file(hf_hub_download(
+    "ByteDance/Hyper-SD", "Hyper-SDXL-1step-Unet.safetensors"), device="cuda")
+pipe.unet.load_state_dict(unet_state)
+pipe.scheduler = LCMScheduler.from_config(
+    pipe.scheduler.config, timestep_spacing="trailing")
+
 print('Loading Finished!')
 
 # Gradio UI