Skip to content

Commit 38e3c48

Browse files
authored
Apply suggestions from code review
1 parent 6e27c43 commit 38e3c48

File tree

4 files changed

+8
-8
lines changed

4 files changed

+8
-8
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
1818
```python
1919
from diffusers import AutoencoderKLWan
2020

21-
vae = AutoencoderKLWan.from_pretrained("Wan-AI/Wan2.1-T2V-1.3B", subfolder="vae", torch_dtype=torch.float32)
21+
vae = AutoencoderKLWan.from_pretrained("Wan-AI/Wan2.1-T2V-1.3B-Diffusers", subfolder="vae", torch_dtype=torch.float32)
2222
```
2323

2424
## AutoencoderKLWan

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ The model can be loaded with the following code snippet.
1818
```python
1919
from diffusers import WanTransformer3DModel
2020

21-
transformer = WanTransformer3DModel.from_pretrained("Wan-AI/Wan2.1-T2V-1.3B", subfolder="transformer", torch_dtype=torch.bfloat16)
21+
transformer = WanTransformer3DModel.from_pretrained("Wan-AI/Wan2.1-T2V-1.3B-Diffusers", subfolder="transformer", torch_dtype=torch.bfloat16)
2222
```
2323

2424
## WanTransformer3DModel

src/diffusers/pipelines/wan/pipeline_wan.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -47,8 +47,8 @@
4747
>>> from diffusers import AutoencoderKLWan, WanPipeline
4848
>>> from diffusers.utils import export_to_video
4949
50-
>>> # Available models: Wan-AI/Wan2.1-T2V-14B, Wan-AI/Wan2.1-T2V-1.3B
51-
>>> model_id = "Wan-AI/Wan2.1-T2V-14B"
50+
>>> # Available models: Wan-AI/Wan2.1-T2V-14B-Diffusers, Wan-AI/Wan2.1-T2V-1.3B-Diffusers
51+
>>> model_id = "Wan-AI/Wan2.1-T2V-14B-Diffusers"
5252
>>> vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
5353
>>> pipe = WanPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
5454
>>> pipe.to("cuda")
@@ -178,7 +178,7 @@ def encode_prompt(
178178
self,
179179
prompt: Union[str, List[str]],
180180
negative_prompt: Optional[Union[str, List[str]]] = None,
181-
do_classifier_free_guidance: bool = False,
181+
do_classifier_free_guidance: bool = True,
182182
num_videos_per_prompt: int = 1,
183183
prompt_embeds: Optional[torch.Tensor] = None,
184184
negative_prompt_embeds: Optional[torch.Tensor] = None,

src/diffusers/pipelines/wan/pipeline_wan_i2v.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,8 @@
4949
>>> from diffusers import AutoencoderKLWan, WanImageToVideoPipeline
5050
>>> from diffusers.utils import export_to_video, load_image
5151
52-
>>> # Available models: Wan-AI/Wan2.1-I2V-14B-480P, Wan-AI/Wan2.1-I2V-1.3B-720P
53-
>>> model_id = "Wan-AI/Wan2.1-I2V-14B-480P"
52+
>>> # Available models: Wan-AI/Wan2.1-I2V-14B-480P-Diffusers, Wan-AI/Wan2.1-I2V-1.3B-720P-Diffusers
53+
>>> model_id = "Wan-AI/Wan2.1-I2V-14B-480P-Diffusers"
5454
>>> vae = AutoencoderKLWan.from_pretrained(model_id, subfolder="vae", torch_dtype=torch.float32)
5555
>>> pipe = WanImageToVideoPipeline.from_pretrained(model_id, vae=vae, torch_dtype=torch.bfloat16)
5656
>>> pipe.to("cuda")
@@ -211,7 +211,7 @@ def encode_prompt(
211211
self,
212212
prompt: Union[str, List[str]],
213213
negative_prompt: Optional[Union[str, List[str]]] = None,
214-
do_classifier_free_guidance: bool = False,
214+
do_classifier_free_guidance: bool = True,
215215
num_videos_per_prompt: int = 1,
216216
prompt_embeds: Optional[torch.Tensor] = None,
217217
negative_prompt_embeds: Optional[torch.Tensor] = None,

0 commit comments

Comments
 (0)