File size: 1,079 Bytes
e2a8f80 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | """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()
|