[diffusion] fix: fix gen video doc (#14409)
Signed-off-by: Xiaodong Ye <yeahdongcn@gmail.com> Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
@@ -37,15 +37,14 @@ def main():
|
|||||||
num_gpus=1, # Adjust based on your hardware
|
num_gpus=1, # Adjust based on your hardware
|
||||||
)
|
)
|
||||||
|
|
||||||
# Provide a prompt for your video
|
|
||||||
prompt = "A curious raccoon peers through a vibrant field of yellow sunflowers, its eyes wide with interest."
|
|
||||||
|
|
||||||
# Generate the video
|
# Generate the video
|
||||||
video = generator.generate(
|
video = generator.generate(
|
||||||
prompt,
|
sampling_params_kwargs=dict(
|
||||||
return_frames=True, # Also return frames from this call (defaults to False)
|
prompt="A curious raccoon peers through a vibrant field of yellow sunflowers, its eyes wide with interest.",
|
||||||
output_path="my_videos/", # Controls where videos are saved
|
return_frames=True, # Also return frames from this call (defaults to False)
|
||||||
save_output=True
|
output_path="my_videos/", # Controls where videos are saved
|
||||||
|
save_output=True
|
||||||
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|||||||
Reference in New Issue
Block a user