Safetensors
katuni4ka commited on
Commit
85c94ff
·
verified ·
1 Parent(s): d63e3e0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -10
README.md CHANGED
@@ -28,22 +28,23 @@ pip install optimum[openvino]
28
 
29
  2. Run model inference:
30
 
 
31
  ```
32
- from transformers import AutoTokenizer
33
- from optimum.intel.openvino import OVModelForCausalLM
34
 
35
  model_id = "OpenVINO/stable-diffusion-v1-5-fp16-ov"
36
- tokenizer = AutoTokenizer.from_pretrained(model_id)
37
- model = OVModelForCausalLM.from_pretrained(model_id)
38
-
39
- inputs = tokenizer("What is OpenVINO?", return_tensors="pt")
40
 
41
- outputs = model.generate(**inputs, max_length=200)
42
- text = tokenizer.batch_decode(outputs)[0]
43
- print(text)
44
  ```
45
 
46
- 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).
 
 
 
 
 
47
 
48
  ## Limitations
49
 
 
28
 
29
  2. Run model inference:
30
 
31
+
32
  ```
33
+ from optimum.intel.openvino import OVStableDiffusionPipeline
 
34
 
35
  model_id = "OpenVINO/stable-diffusion-v1-5-fp16-ov"
36
+ pipeline = OVStableDiffusionPipeline.from_pretrained(model_id)
 
 
 
37
 
38
+ prompt = "sailing ship in storm by Rembrandt"
39
+ images = pipeline(prompt, num_inference_steps=4).images
 
40
  ```
41
 
42
+ ## Usage examples
43
+
44
+ * [OpenVINO notebooks](https://github.com/openvinotoolkit/openvino_notebooks):
45
+ - [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)
46
+ * [OpenVINO GenAI](https://github.com/openvinotoolkit/openvino.genai):
47
+ - [C++ image generation pipeline](https://github.com/openvinotoolkit/openvino.genai/tree/master/image_generation/stable_diffusion_1_5/cpp)
48
 
49
  ## Limitations
50