Manjushri commited on
Commit
d6e0495
1 Parent(s): 7597a8f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,10 +20,10 @@ if torch.cuda.is_available():
20
  else:
21
  pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
22
  pipe = pipe.to(device)
23
- pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
24
  refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", use_safetensors=True)
25
  refiner = refiner.to(device)
26
- refiner.unet = torch.compile(refiner.unet, mode="reduce-overhead", fullgraph=True)
27
 
28
  def genie (prompt, negative_prompt, height, width, scale, steps, seed, prompt_2, negative_prompt_2, high_noise_frac):
29
  generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)
 
20
  else:
21
  pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True)
22
  pipe = pipe.to(device)
23
+ #pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
24
  refiner = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", use_safetensors=True)
25
  refiner = refiner.to(device)
26
+ #refiner.unet = torch.compile(refiner.unet, mode="reduce-overhead", fullgraph=True)
27
 
28
  def genie (prompt, negative_prompt, height, width, scale, steps, seed, prompt_2, negative_prompt_2, high_noise_frac):
29
  generator = np.random.seed(0) if seed == 0 else torch.manual_seed(seed)