Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ def generate_image(prompt: str, inference_steps: int = 30, prng_seed: int = 0):
|
|
18 |
height=128,
|
19 |
width=128,
|
20 |
num_inference_steps=int(inference_steps),
|
21 |
-
jit=
|
22 |
).images
|
23 |
|
24 |
pil_imgs = pipeline.numpy_to_pil(images)
|
|
|
18 |
height=128,
|
19 |
width=128,
|
20 |
num_inference_steps=int(inference_steps),
|
21 |
+
jit=True,
|
22 |
).images
|
23 |
|
24 |
pil_imgs = pipeline.numpy_to_pil(images)
|