HelloSun commited on
Commit
e442d7e
·
verified ·
1 Parent(s): e7bb88b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -32,7 +32,7 @@ pipe.compile()
32
  prompt=""
33
  negative_prompt=""
34
 
35
- def infer(prompt,negative_prompt,num_inference_steps):
36
 
37
  image = pipe(
38
  prompt = prompt,
@@ -40,7 +40,7 @@ def infer(prompt,negative_prompt,num_inference_steps):
40
  width = HIGH,
41
  height = WIDTH,
42
  guidance_scale=1.0,
43
- num_inference_steps=num_inference_steps,
44
  num_images_per_prompt=1,
45
  ).images[0]
46
 
 
32
  prompt=""
33
  negative_prompt=""
34
 
35
+ def infer(prompt,negative_prompt):
36
 
37
  image = pipe(
38
  prompt = prompt,
 
40
  width = HIGH,
41
  height = WIDTH,
42
  guidance_scale=1.0,
43
+ num_inference_steps=4,
44
  num_images_per_prompt=1,
45
  ).images[0]
46