Spaces:
Build error
Build error
Commit
·
86ea458
1
Parent(s):
aad0875
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ from optimum.intel.openvino import OVStableDiffusionPipeline
|
|
| 3 |
from diffusers.training_utils import set_seed
|
| 4 |
|
| 5 |
quantized_pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/Stable-Diffusion-Pokemon-en-quantized", compile=False)
|
| 6 |
-
quantized_pipe.reshape(batch_size=1, height=
|
| 7 |
quantized_pipe.compile()
|
| 8 |
|
| 9 |
prompt = "cartoon bird"
|
|
@@ -24,5 +24,6 @@ gr.Interface(
|
|
| 24 |
outputs=gr.outputs.Image(type="pil", label="Generated Image"),
|
| 25 |
title="OpenVINO-optimized Stable Diffusion",
|
| 26 |
description="",
|
|
|
|
| 27 |
theme="huggingface",
|
| 28 |
).launch()
|
|
|
|
| 3 |
from diffusers.training_utils import set_seed
|
| 4 |
|
| 5 |
quantized_pipe = OVStableDiffusionPipeline.from_pretrained("OpenVINO/Stable-Diffusion-Pokemon-en-quantized", compile=False)
|
| 6 |
+
quantized_pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
|
| 7 |
quantized_pipe.compile()
|
| 8 |
|
| 9 |
prompt = "cartoon bird"
|
|
|
|
| 24 |
outputs=gr.outputs.Image(type="pil", label="Generated Image"),
|
| 25 |
title="OpenVINO-optimized Stable Diffusion",
|
| 26 |
description="",
|
| 27 |
+
examples=examples,
|
| 28 |
theme="huggingface",
|
| 29 |
).launch()
|