@@ -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