from optimum.intel.openvino import OVDiffusionPipeline

model_id = "OpenVINO/FLUX.1-schnell-int4-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-int4-ov"
model_path = "FLUX.1-schnell-int4-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])