ChenyangSi commited on
Commit
693cf66
1 Parent(s): 7f15b2e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -49,7 +49,7 @@ def infer(prompt, sd_options, seed, b1, b2, s1, s2):
49
 
50
  torch.manual_seed(seed)
51
  print("Generating SD:")
52
- sd_image = pip(prompt, num_inference_steps=25).images[0]
53
  sd_image_prev = sd_image
54
  else:
55
  sd_image = sd_image_prev
@@ -60,7 +60,7 @@ def infer(prompt, sd_options, seed, b1, b2, s1, s2):
60
 
61
  torch.manual_seed(seed)
62
  print("Generating FreeU:")
63
- freeu_image = pip(prompt, num_inference_steps=25).images[0]
64
 
65
  # First SD, then freeu
66
  images = [sd_image, freeu_image]
 
49
 
50
  torch.manual_seed(seed)
51
  print("Generating SD:")
52
+ sd_image = pip(prompt).images[0]
53
  sd_image_prev = sd_image
54
  else:
55
  sd_image = sd_image_prev
 
60
 
61
  torch.manual_seed(seed)
62
  print("Generating FreeU:")
63
+ freeu_image = pip(prompt).images[0]
64
 
65
  # First SD, then freeu
66
  images = [sd_image, freeu_image]