Skip to content

Commit 2d65aa2

Browse files
committed
doc updates
1 parent 3ec1896 commit 2d65aa2

File tree

2 files changed

+18
-0
lines changed

2 files changed

+18
-0
lines changed

docs/source/en/api/models/hidream_image_transformer.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,14 @@ specific language governing permissions and limitations under the License. -->
1313

1414
A Transformer model for image-like data from [HiDream-I1](https://huggingface.co/HiDream-ai).
1515

16+
The model can be loaded with the following code snippet.
17+
18+
```python
19+
from diffusers import HiDreamImageTransformer2DModel
20+
21+
transformer = HiDreamImageTransformer2DModel.from_pretrained("HiDream-ai/HiDream-I1-Full", subfolder="transformer", torch_dtype=torch.bfloat16)
22+
```
23+
1624
## HiDreamImageTransformer2DModel
1725

1826
[[autodoc]] HiDreamImageTransformer2DModel

docs/source/en/api/pipelines/hidream.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,16 @@ Make sure to check out the Schedulers [guide](../../using-diffusers/schedulers)
2222

2323
</Tip>
2424

25+
## Available models
26+
27+
The following models are available for the [`HiDreamImagePipeline`](text-to-image) pipeline:
28+
29+
| Model name | Description |
30+
|:---|:---|
31+
| [`HiDream-ai/HiDream-I1-Full`](https://huggingface.co/HiDream-ai/HiDream-I1-Full) | - |
32+
| [`HiDream-ai/HiDream-I1-Dev`](https://huggingface.co/HiDream-ai/HiDream-I1-Dev) | - |
33+
| [`HiDream-ai/HiDream-I1-Fast`](https://huggingface.co/HiDream-ai/HiDream-I1-Fast) | - |
34+
2535
## HiDreamImagePipeline
2636

2737
[[autodoc]] HiDreamImagePipeline

0 commit comments

Comments
 (0)