multimodalart HF staff commited on
Commit
3509a66
1 Parent(s): 6f4d50e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(base_model_id, 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
 
 
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