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()
|