Skip to content

Commit e8312e7

Browse files
authored
[BUG] fixed WAN docstring (#11226)
update
1 parent 7986834 commit e8312e7

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

src/diffusers/pipelines/wan/pipeline_wan.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -417,7 +417,7 @@ def __call__(
417417
prompt_embeds (`torch.Tensor`, *optional*):
418418
Pre-generated text embeddings. Can be used to easily tweak text inputs (prompt weighting). If not
419419
provided, text embeddings are generated from the `prompt` input argument.
420-
output_type (`str`, *optional*, defaults to `"pil"`):
420+
output_type (`str`, *optional*, defaults to `"np"`):
421421
The output format of the generated image. Choose between `PIL.Image` or `np.array`.
422422
return_dict (`bool`, *optional*, defaults to `True`):
423423
Whether or not to return a [`WanPipelineOutput`] instead of a plain tuple.

src/diffusers/pipelines/wan/pipeline_wan_i2v.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -545,7 +545,7 @@ def __call__(
545545
image_embeds (`torch.Tensor`, *optional*):
546546
Pre-generated image embeddings. Can be used to easily tweak image inputs (weighting). If not provided,
547547
image embeddings are generated from the `image` input argument.
548-
output_type (`str`, *optional*, defaults to `"pil"`):
548+
output_type (`str`, *optional*, defaults to `"np"`):
549549
The output format of the generated image. Choose between `PIL.Image` or `np.array`.
550550
return_dict (`bool`, *optional*, defaults to `True`):
551551
Whether or not to return a [`WanPipelineOutput`] instead of a plain tuple.

src/diffusers/pipelines/wan/pipeline_wan_video2video.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -542,7 +542,7 @@ def __call__(
542542
prompt_embeds (`torch.Tensor`, *optional*):
543543
Pre-generated text embeddings. Can be used to easily tweak text inputs (prompt weighting). If not
544544
provided, text embeddings are generated from the `prompt` input argument.
545-
output_type (`str`, *optional*, defaults to `"pil"`):
545+
output_type (`str`, *optional*, defaults to `"np"`):
546546
The output format of the generated image. Choose between `PIL.Image` or `np.array`.
547547
return_dict (`bool`, *optional*, defaults to `True`):
548548
Whether or not to return a [`WanPipelineOutput`] instead of a plain tuple.

0 commit comments

Comments
 (0)