Skip to content

Commit d791908

Browse files
authored
Apply suggestions from code review
1 parent 6239a6a commit d791908

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/diffusers/pipelines/mochi/pipeline_mochi.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -477,7 +477,7 @@ def __call__(
477477
negative_prompt: Optional[Union[str, List[str]]] = None,
478478
height: Optional[int] = None,
479479
width: Optional[int] = None,
480-
num_frames: int = 16,
480+
num_frames: int = 19,
481481
num_inference_steps: int = 28,
482482
timesteps: List[int] = None,
483483
guidance_scale: float = 4.5,
@@ -501,11 +501,11 @@ def __call__(
501501
prompt (`str` or `List[str]`, *optional*):
502502
The prompt or prompts to guide the image generation. If not defined, one has to pass `prompt_embeds`.
503503
instead.
504-
height (`int`, *optional*, defaults to self.unet.config.sample_size * self.vae_scale_factor):
505-
The height in pixels of the generated image. This is set to 1024 by default for the best results.
506-
width (`int`, *optional*, defaults to self.unet.config.sample_size * self.vae_scale_factor):
507-
The width in pixels of the generated image. This is set to 1024 by default for the best results.
508-
num_frames (`int`, defaults to 16):
504+
height (`int`, *optional*, defaults to `self.default_height`):
505+
The height in pixels of the generated image. This is set to 480 by default for the best results.
506+
width (`int`, *optional*, defaults to `self.default_width`):
507+
The width in pixels of the generated image. This is set to 848 by default for the best results.
508+
num_frames (`int`, defaults to `19`):
509509
The number of video frames to generate
510510
num_inference_steps (`int`, *optional*, defaults to 50):
511511
The number of denoising steps. More denoising steps usually lead to a higher quality image at the

0 commit comments

Comments
 (0)