feat: generation options for infer.py, and split out training deps

infer.py could only render a random example from examples/input_params:
there was no way to pass your own prompt, lyrics, duration or seed, so
using it for anything specific meant writing a separate script. Add
--prompt, --lyrics/--lyrics_file, --duration, --steps, --guidance_scale,
--scheduler, --cfg_type, --omega_scale, --seed and --format alongside the
existing runtime flags. Without --prompt the old random-example behaviour
is kept, so existing invocations are unaffected.

Also move the training-only packages out of the default install.
datasets, pytorch_lightning, matplotlib, tensorboard and tensorboardX are
imported by trainer.py and convert2hf_dataset.py, never on the inference
path, but every user was installing them — and datasets==3.4.1 is a hard
pin that drags constraints onto huggingface-hub. They now live in
requirements-train.txt behind the existing (previously ineffective)
"train" extra: pip install -e ".[train]".

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>

This commit is contained in:

Leonid Pershin

2026-09-08 20:31:07 +03:00

co-authored by

Claude Opus 5

parent c7953dc4e0

commit e9ea6b9bab

5 changed files with 145 additions and 40 deletions

requirements-train.txt

View File

@@ -0,0 +1,7 @@
 # Training-only dependencies (trainer.py, convert2hf_dataset.py).
 # Install with: pip install -e ".[train]"
 datasets==3.4.1
 pytorch_lightning==2.5.1
 matplotlib==3.10.1
 tensorboard
 tensorboardX