Update app.py
Browse files
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,
|
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=
|
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 |
|