stable-diffusion-2-1-int8-ov
- Model creator: Stability AI
- Original model: stable-diffusion-2-1
Description
This is stable-diffusion-2-1 model converted to the OpenVINO™ IR (Intermediate Representation) format with weights compressed to INT8 by NNCF.
For more information on quantization, check the OpenVINO model optimization guide.
Running Model Inference
- Install packages required for using Optimum Intel integration with the OpenVINO backend:
pip install optimum[openvino]
- Run model inference:
from optimum.intel.openvino import OVStableDiffusionPipeline
model_id = "OpenVINO/stable-diffusion-2-1-int8-ov"
pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile=False)
pipe.reshape(batch_size=1, height=512, width=512, num_images_per_prompt=1)
pipe.compile()
prompt = "sailing ship in storm by Rembrandt"
output = pipe(prompt, num_inference_steps=50, output_type="pil")
output.images[0].save("result.png")
For more examples and possible optimizations, refer to the OpenVINO Large Language Model Inference Guide.
You can find more detailed usage examples in OpenVINO Notebooks:
Limitations
Check the original model card for limitations.
Legal information
The original model is distributed under openrail++ license. More details can be found in original model card.
Disclaimer
Intel is committed to respecting human rights and avoiding causing or contributing to adverse impacts on human rights. See Intel’s Global Human Rights Principles. Intel’s products and software are intended only to be used in applications that do not cause or contribute to adverse impacts on human rights.