Safetensors
katuni4ka commited on
Commit
646c081
·
verified ·
1 Parent(s): 4ed4826

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -10
README.md CHANGED
@@ -40,21 +40,23 @@ pip install optimum[openvino]
40
  2. Run model inference:
41
 
42
  ```
43
- from transformers import AutoTokenizer
44
- from optimum.intel.openvino import OVModelForCausalLM
45
 
46
  model_id = "OpenVINO/stable-diffusion-v1-5-int8-ov"
47
- tokenizer = AutoTokenizer.from_pretrained(model_id)
48
- model = OVModelForCausalLM.from_pretrained(model_id)
49
-
50
- inputs = tokenizer("What is OpenVINO?", return_tensors="pt")
51
 
52
- outputs = model.generate(**inputs, max_length=200)
53
- text = tokenizer.batch_decode(outputs)[0]
54
- print(text)
55
  ```
56
 
57
- For more examples and possible optimizations, refer to the [OpenVINO Large Language Model Inference Guide](https://docs.openvino.ai/2024/learn-openvino/llm_inference_guide.html).
 
 
 
 
 
 
58
 
59
  ## Limitations
60
 
 
40
  2. Run model inference:
41
 
42
  ```
43
+ ```
44
+ from optimum.intel.openvino import OVStableDiffusionPipeline
45
 
46
  model_id = "OpenVINO/stable-diffusion-v1-5-int8-ov"
47
+ pipeline = OVStableDiffusionPipeline.from_pretrained(model_id)
 
 
 
48
 
49
+ prompt = "sailing ship in storm by Rembrandt"
50
+ images = pipeline(prompt, num_inference_steps=4).images
 
51
  ```
52
 
53
+ ## Usage examples
54
+
55
+ * [OpenVINO notebooks](https://github.com/openvinotoolkit/openvino_notebooks):
56
+ - [Latent Consistency Model using Optimum-Intel OpenVINO](https://github.com/openvinotoolkit/openvino_notebooks/blob/latest/notebooks/stable-diffusion-text-to-image/stable-diffusion-text-to-image.ipynb)
57
+ * [OpenVINO GenAI](https://github.com/openvinotoolkit/openvino.genai):
58
+ - [C++ image generation pipeline](https://github.com/openvinotoolkit/openvino.genai/tree/master/image_generation/stable_diffusion_1_5/cpp)
59
+ ```
60
 
61
  ## Limitations
62