commit files to HF hub
Browse files
README.md
CHANGED
@@ -27,7 +27,7 @@ model from the Hugging Face hub (if it has not been downloaded before) and loads
|
|
27 |
```python
|
28 |
from optimum.intel.openvino import OVStableDiffusionPipeline
|
29 |
|
30 |
-
stable_diffusion = OVStableDiffusionPipeline.from_pretrained("naclbit
|
31 |
images = stable_diffusion("a random image").images
|
32 |
```
|
33 |
|
@@ -49,7 +49,7 @@ height = 256
|
|
49 |
width = 256
|
50 |
|
51 |
# load the model and reshape to static shapes for faster inference
|
52 |
-
model_id = "naclbit
|
53 |
stable_diffusion = OVStableDiffusionPipeline.from_pretrained(model_id, compile=False)
|
54 |
stable_diffusion.reshape( batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images_per_prompt)
|
55 |
stable_diffusion.compile()
|
|
|
27 |
```python
|
28 |
from optimum.intel.openvino import OVStableDiffusionPipeline
|
29 |
|
30 |
+
stable_diffusion = OVStableDiffusionPipeline.from_pretrained("helenai/naclbit-trinart_stable_diffusion_v2-ov")
|
31 |
images = stable_diffusion("a random image").images
|
32 |
```
|
33 |
|
|
|
49 |
width = 256
|
50 |
|
51 |
# load the model and reshape to static shapes for faster inference
|
52 |
+
model_id = "helenai/naclbit-trinart_stable_diffusion_v2-ov"
|
53 |
stable_diffusion = OVStableDiffusionPipeline.from_pretrained(model_id, compile=False)
|
54 |
stable_diffusion.reshape( batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images_per_prompt)
|
55 |
stable_diffusion.compile()
|