Skip to content

Commit 785e230

Browse files
Update pipeline_cogview4_control.py
1 parent 8cdd36f commit 785e230

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/diffusers/pipelines/cogview4/pipeline_cogview4_control.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -523,7 +523,7 @@ def __call__(
523523
The output format of the generate image. Choose between
524524
[PIL](https://pillow.readthedocs.io/en/stable/): `PIL.Image.Image` or `np.array`.
525525
return_dict (`bool`, *optional*, defaults to `True`):
526-
Whether or not to return a [`~pipelines.stable_diffusion_xl.StableDiffusionXLPipelineOutput`] instead
526+
Whether or not to return a [`~pipelines.pipeline_CogView4.CogView4PipelineOutput`] instead
527527
of a plain tuple.
528528
attention_kwargs (`dict`, *optional*):
529529
A kwargs dictionary that if specified is passed along to the `AttentionProcessor` as defined under
@@ -541,8 +541,6 @@ def __call__(
541541
max_sequence_length (`int`, defaults to `224`):
542542
Maximum sequence length in encoded prompt. Can be set to other values but may lead to poorer results.
543543
544-
Examples:
545-
546544
Returns:
547545
[`~pipelines.cogview4.pipeline_CogView4.CogView4PipelineOutput`] or `tuple`:
548546
[`~pipelines.cogview4.pipeline_CogView4.CogView4PipelineOutput`] if `return_dict` is True, otherwise a

0 commit comments

Comments
 (0)