Skip to content

Commit 9f2bff5

Browse files
authored
[svd] fix noise_aug_strength type in svd pipe (huggingface#6389)
1 parent 0cb9271 commit 9f2bff5

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/diffusers/pipelines/stable_video_diffusion/pipeline_stable_video_diffusion.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -311,7 +311,7 @@ def __call__(
311311
max_guidance_scale: float = 3.0,
312312
fps: int = 7,
313313
motion_bucket_id: int = 127,
314-
noise_aug_strength: int = 0.02,
314+
noise_aug_strength: float = 0.02,
315315
decode_chunk_size: Optional[int] = None,
316316
num_videos_per_prompt: Optional[int] = 1,
317317
generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
@@ -346,7 +346,7 @@ def __call__(
346346
Note that Stable Diffusion Video's UNet was micro-conditioned on fps-1 during training.
347347
motion_bucket_id (`int`, *optional*, defaults to 127):
348348
The motion bucket ID. Used as conditioning for the generation. The higher the number the more motion will be in the video.
349-
noise_aug_strength (`int`, *optional*, defaults to 0.02):
349+
noise_aug_strength (`float`, *optional*, defaults to 0.02):
350350
The amount of noise added to the init image, the higher it is the less the video will look like the init image. Increase it for more motion.
351351
decode_chunk_size (`int`, *optional*):
352352
The number of frames to decode at a time. The higher the chunk size, the higher the temporal consistency

0 commit comments

Comments
 (0)