HelloSun commited on
Commit
188c7ea
·
verified ·
1 Parent(s): 52ee8e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,7 +26,7 @@ pipe.reshape( batch_size=-1, height=512, width=512, num_images_per_prompt=1)
26
  pipe.compile()
27
 
28
 
29
- def infer(prompt,negative_prompt, step):
30
 
31
  image = pipe(
32
  prompt = prompt,
@@ -34,7 +34,7 @@ def infer(prompt,negative_prompt, step):
34
  width = 512,
35
  height = 512,
36
  guidance_scale=1.0,
37
- num_inference_steps=8,
38
  num_images_per_prompt=1,
39
  ).images[0]
40
 
 
26
  pipe.compile()
27
 
28
 
29
+ def infer(prompt,negative_prompt, num_inference_steps):
30
 
31
  image = pipe(
32
  prompt = prompt,
 
34
  width = 512,
35
  height = 512,
36
  guidance_scale=1.0,
37
+ num_inference_steps=step,
38
  num_images_per_prompt=1,
39
  ).images[0]
40