Manjushri commited on
Commit
f73e253
·
1 Parent(s): ece7638

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ pipe = pipe.to(device)
16
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
17
 
18
  refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9")
19
- refiner = pipe.to(device)
20
- #refiner.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
21
 
22
  def genie (prompt, negative_prompt, scale, steps, seed):
23
  generator = torch.Generator(device=device).manual_seed(seed)
 
16
  pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
17
 
18
  refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9")
19
+ refiner = refiner.to(device)
20
+ refiner.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
21
 
22
  def genie (prompt, negative_prompt, scale, steps, seed):
23
  generator = torch.Generator(device=device).manual_seed(seed)