Davids048andClaude Opus 5.5
f571621ae7
Every example calls VideoGenerator.generate with a nested request instead of generate_video keywords, and passes settings outside the from_pretrained convenience keywords through VideoGenerator.from_config at their typed paths. Each request and generator config equals what the keyword call built. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_012M91wnVFmPEvJ9h39r5BH7
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""MMAudio large-44k-v2 video-to-audio example."""
|
|
|
|
import argparse
|
|
import os
|
|
|
|
from fastvideo import VideoGenerator
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
parser.add_argument("--video-path", required=True)
|
|
parser.add_argument("--output-path", default="outputs_audio/mmaudio.wav")
|
|
parser.add_argument("--duration-seconds", type=float, default=8.0)
|
|
parser.add_argument("--prompt", default="")
|
|
parser.add_argument("--negative-prompt", default="music")
|
|
return parser.parse_args()
|
|
|
|
|
|
def main() -> None:
|
|
args = parse_args()
|
|
generator = VideoGenerator.from_config({
|
|
"model_path": os.environ.get(
|
|
"MMAUDIO_MODEL_PATH",
|
|
"converted_weights/mmaudio/large_44k_v2",
|
|
),
|
|
"engine": {"num_gpus": 1},
|
|
"pipeline": {"workload_type": "v2a"},
|
|
})
|
|
result = generator.generate({
|
|
"prompt": args.prompt,
|
|
"negative_prompt": args.negative_prompt,
|
|
"inputs": {"video_path": args.video_path},
|
|
"output": {"output_path": args.output_path, "save_video": True, "return_frames": False},
|
|
"extensions": {"audio_end_in_s": args.duration_seconds},
|
|
})
|
|
print(result.video_path)
|
|
generator.shutdown()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|