hsuwill000 commited on
Commit
5c31923
·
verified ·
1 Parent(s): 0292d88

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,15 +25,15 @@ pipe.compile()
25
 
26
  prompt = ""
27
  negative_prompt = "EasyNegative, "
28
-
29
- def infer(prompt, negative_prompt):
30
  image = pipe(
31
  prompt=prompt,
32
  negative_prompt=negative_prompt,
33
  width=WIDTH,
34
  height=HIGH,
35
  guidance_scale=1.0,
36
- num_inference_steps=8,
37
  num_images_per_prompt=1,
38
  ).images[0]
39
 
 
25
 
26
  prompt = ""
27
  negative_prompt = "EasyNegative, "
28
+ num_inference_steps = 4
29
+ def infer(prompt, negative_prompt, num_inference_steps):
30
  image = pipe(
31
  prompt=prompt,
32
  negative_prompt=negative_prompt,
33
  width=WIDTH,
34
  height=HIGH,
35
  guidance_scale=1.0,
36
+ num_inference_steps=num_inference_steps,
37
  num_images_per_prompt=1,
38
  ).images[0]
39