diff --git a/src/diffusers/pipelines/mochi/pipeline_output.py b/src/diffusers/pipelines/mochi/pipeline_output.py index cc14372794..d15827bc00 100644 --- a/src/diffusers/pipelines/mochi/pipeline_output.py +++ b/src/diffusers/pipelines/mochi/pipeline_output.py @@ -8,7 +8,7 @@ from diffusers.utils import BaseOutput @dataclass class MochiPipelineOutput(BaseOutput): r""" - Output class for CogVideo pipelines. + Output class for Mochi pipelines. Args: frames (`torch.Tensor`, `np.ndarray`, or List[List[PIL.Image.Image]]):