From 3ff39e8e860a952096054a14af8d4125668d5801 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Tue, 4 Jun 2024 12:18:53 +0400 Subject: [PATCH] [HunyuanDiT] minor docs changes in hunyuandit (#8395) minor docs changes in hunyuandit --- docs/source/en/_toctree.yml | 2 +- .../{hunyuan_transformer_2d.md => hunyuan_transformer2d.md} | 0 docs/source/en/api/pipelines/hunyuandit.md | 2 +- src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py | 4 +++- 4 files changed, 5 insertions(+), 3 deletions(-) rename docs/source/en/api/models/{hunyuan_transformer_2d.md => hunyuan_transformer2d.md} (100%) diff --git a/docs/source/en/_toctree.yml b/docs/source/en/_toctree.yml index ed4adc9bb4..1ecf82a7a6 100644 --- a/docs/source/en/_toctree.yml +++ b/docs/source/en/_toctree.yml @@ -242,7 +242,7 @@ title: PixArtTransformer2DModel - local: api/models/dit_transformer2d title: DiTTransformer2DModel - - local: api/models/hunyuan_transformer_2d + - local: api/models/hunyuan_transformer2d title: HunyuanDiT2DModel - local: api/models/transformer_temporal title: TransformerTemporalModel diff --git a/docs/source/en/api/models/hunyuan_transformer_2d.md b/docs/source/en/api/models/hunyuan_transformer2d.md similarity index 100% rename from docs/source/en/api/models/hunyuan_transformer_2d.md rename to docs/source/en/api/models/hunyuan_transformer2d.md diff --git a/docs/source/en/api/pipelines/hunyuandit.md b/docs/source/en/api/pipelines/hunyuandit.md index 2662ce3a81..f777d42e9b 100644 --- a/docs/source/en/api/pipelines/hunyuandit.md +++ b/docs/source/en/api/pipelines/hunyuandit.md @@ -13,7 +13,7 @@ specific language governing permissions and limitations under the License. # Hunyuan-DiT ![chinese elements understanding](https://github.com/gnobitab/diffusers-hunyuan/assets/1157982/39b99036-c3cb-4f16-bb1a-40ec25eda573) -[Hunyuan-DiT : A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understanding](https://arxiv.org/abs/2405.08748)] from Tencent Hunyuan. +[Hunyuan-DiT : A Powerful Multi-Resolution Diffusion Transformer with Fine-Grained Chinese Understanding](https://arxiv.org/abs/2405.08748) from Tencent Hunyuan. The abstract from the paper is: diff --git a/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py b/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py index 835a553c12..46e19f79cc 100644 --- a/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py +++ b/src/diffusers/pipelines/hunyuandit/pipeline_hunyuandit.py @@ -52,7 +52,9 @@ EXAMPLE_DOC_STRING = """ >>> import torch >>> from diffusers import HunyuanDiTPipeline - >>> pipe = HunyuanDiTPipeline.from_pretrained("Tencent-Hunyuan/HunyuanDiT", torch_dtype=torch.float16) + >>> pipe = HunyuanDiTPipeline.from_pretrained( + ... "Tencent-Hunyuan/HunyuanDiT-Diffusers", torch_dtype=torch.float16 + ... ) >>> pipe.to("cuda") >>> # You may also use English prompt as HunyuanDiT supports both English and Chinese