Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
3509a66
1
Parent(s):
6f4d50e
Update app.py
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ ckpt_name = "Hyper-SDXL-1step-Unet.safetensors"
|
|
29 |
|
30 |
unet = UNet2DConditionModel.from_config(base, subfolder="unet").to("cuda", torch.float16)
|
31 |
unet.load_state_dict(load_file(hf_hub_download(repo_name, ckpt_name), device="cuda"))
|
32 |
-
pipe_hyper = StableDiffusionXLPipeline.from_pretrained(
|
33 |
pipe_hyper.scheduler = LCMScheduler.from_config(pipe_hyper.scheduler.config)
|
34 |
pipe_hyper.to("cuda")
|
35 |
|
|
|
29 |
|
30 |
unet = UNet2DConditionModel.from_config(base, subfolder="unet").to("cuda", torch.float16)
|
31 |
unet.load_state_dict(load_file(hf_hub_download(repo_name, ckpt_name), device="cuda"))
|
32 |
+
pipe_hyper = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
|
33 |
pipe_hyper.scheduler = LCMScheduler.from_config(pipe_hyper.scheduler.config)
|
34 |
pipe_hyper.to("cuda")
|
35 |
|