optimized_inference.py

"""Lance Ditto avec une taille source plafonnée, sans modifier le code du dépôt Ditto."""
from __future__ import annotations

import argparse
import sys
from pathlib import Path


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument("--ditto-dir", required=True)
    parser.add_argument("--data-root", required=True)
    parser.add_argument("--cfg-pkl", required=True)
    parser.add_argument("--audio-path", required=True)
    parser.add_argument("--source-path", required=True)
    parser.add_argument("--output-path", required=True)
    parser.add_argument("--max-size", type=int, default=640)
    args = parser.parse_args()

    ditto_dir = Path(args.ditto_dir).resolve()
    sys.path.insert(0, str(ditto_dir))
    from inference import run
    from stream_pipeline_offline import StreamSDK

    sdk = StreamSDK(args.cfg_pkl, args.data_root)
    run(
        sdk,
        args.audio_path,
        args.source_path,
        args.output_path,
        {"setup_kwargs": {"max_size": args.max_size}},
    )


if __name__ == "__main__":
    main()