|
from optimum.intel.openvino.modeling_diffusion import OVStableDiffusionPipeline |
|
|
|
batch_size = 1 |
|
num_images_per_prompt = 1 |
|
height = 256 |
|
width = 256 |
|
|
|
|
|
model_id = "helenai/CompVis-stable-diffusion-v1-4-ov" |
|
stable_diffusion = OVStableDiffusionPipeline.from_pretrained(model_id, compile=False) |
|
stable_diffusion.reshape( batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images_per_prompt) |
|
stable_diffusion.compile() |
|
|
|
|
|
prompt = "a random image" |
|
images = stable_diffusion(prompt, height=height, width=width, num_images_per_prompt=num_images_per_prompt).images |
|
images[0].save("result.png") |
|
|
|
|