Image Generation
Collection
Collection of OpenVINO optimized models for image generation
•
7 items
•
Updated
•
1
This is black-forest-labs/FLUX.1-schnell model converted to the OpenVINO™ IR (Intermediate Representation) format.
The provided OpenVINO™ IR model is compatible with:
pip install optimum[openvino]
from optimum.intel.openvino import OVDiffusionPipeline
model_id = "OpenVINO/FLUX.1-schnell-fp16-ov"
pipeline = OVDiffusionPipeline.from_pretrained(model_id)
prompt = "A cat holding a sign that says hello world"
images = pipeline(prompt, guidance_scale=0.0, num_inference_steps=4).images
pip install huggingface_hub
pip install -U --pre --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly openvino openvino-tokenizers openvino-genai
import huggingface_hub as hf_hub
model_id = "OpenVINO/FLUX.1-schnell-fp16-ov"
model_path = "FLUX.1-schnell-fp16-ov"
hf_hub.snapshot_download(model_id, local_dir=model_path)
import openvino_genai as ov_genai
from PIL import Image
device = "CPU"
pipe = ov_genai.Text2ImagePipeline(model_path, device)
prompt = "A cat holding a sign that says hello world"
image_tensor = pipe.generate(prompt, guidance_scale=0.0, 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
Check the original model card for limitations.
The original model is distributed under Apache 2.0 license. More details can be found in model card.
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.
Base model
black-forest-labs/FLUX.1-schnell