PeterL1n commited on
Commit
ce4a9a7
1 Parent(s): 33a4306

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -22,8 +22,7 @@ opts = {
22
  step_loaded = None
23
  # unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device, dtype)
24
  # unet.load_state_dict(load_file(hf_hub_download(repo, opts["4 Steps"][0]), device=device))
25
- unet = None
26
- pipe = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=dtype, variant="fp16").to(device)
27
 
28
  @spaces.GPU(enable_queue=True)
29
  def generate_image(prompt, option):
 
22
  step_loaded = None
23
  # unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device, dtype)
24
  # unet.load_state_dict(load_file(hf_hub_download(repo, opts["4 Steps"][0]), device=device))
25
+ pipe = StableDiffusionXLPipeline.from_pretrained(base, torch_dtype=dtype, variant="fp16").to(device)
 
26
 
27
  @spaces.GPU(enable_queue=True)
28
  def generate_image(prompt, option):