Safetensors
katuni4ka's picture
Update README.md
68eca30 verified
metadata
license: creativeml-openrail-m

stable-diffusion-v1-5-int8-ov

Description

This is stable-diffusion-v1-5/stable-diffusion-v1-5 model converted to the OpenVINO™ IR (Intermediate Representation) format with weights compressed to INT8 by NNCF.

Quantization Parameters

Weight compression was performed using nncf.compress_weights with the following parameters:

  • mode: INT8_ASYM
  • ratio: 1.0

For more information on quantization, check the OpenVINO model optimization guide.

Compatibility

The provided OpenVINO™ IR model is compatible with:

  • OpenVINO version 2024.6.0 and higher
  • Optimum Intel 1.21.0 and higher

Running Model Inference with Optimum Intel

  1. Install packages required for using Optimum Intel integration with the OpenVINO backend:
pip install optimum[openvino]
  1. Run model inference:
from optimum.intel.openvino import OVDiffusionPipeline

model_id = "OpenVINO/stable-diffusion-v1-5-int8-ov"
pipeline = OVDiffusionPipeline.from_pretrained(model_id)

prompt = "sailing ship in storm by Rembrandt"
images = pipeline(prompt, num_inference_steps=4).images

Running Model Inference with OpenVINO GenAI

  1. Install packages required for using OpenVINO GenAI.
pip install huggingface_hub
pip install -U --pre --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly openvino openvino-tokenizers openvino-genai
  1. Download model from HuggingFace Hub
import huggingface_hub as hf_hub

model_id = "OpenVINO/stable-diffusion-v1-5-int8-ov"
model_path = "stable-diffusion-v1-5-int8-ov"

hf_hub.snapshot_download(model_id, local_dir=model_path)
  1. Run model inference:
import openvino_genai as ov_genai
from PIL import Image

device = "CPU"
pipe = ov_genai.Text2ImagePipeline(model_path, device)

prompt = "sailing ship in storm by Rembrandt"
image_tensor = pipe.generate(prompt, num_inference_steps=4)
image = Image.fromarray(image_tensor.data[0])

More GenAI usage examples can be found in OpenVINO GenAI library docs and samples

Limitations

Check the original model card for limitations.

Legal information

The original model is distributed under creativeml-openrail-m license. More details can be found in stable-diffusion-v1-5/stable-diffusion-v1-5.

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.