Broken docker build

[DarkTyger]

Issue

Terminal 1:

$ uname -a
Linux hostname 6.15.2-arch1-1 #1 SMP PREEMPT_DYNAMIC Tue, 10 Jun 2025 21:32:33 +0000 x86_64 GNU/Linux
$ docker --version
Docker version 28.2.2, build e6534b4eb7
$ mkdir $HOME/tts/
$ cd $HOME/tts/
$ git clone https://github.com/bytedance/MegaTTS3 mega
$ cd mega
$ mkdir checkpoints
$ cd checkpoints
$ ls ~/archive/*zip | xargs -n 1 basename
aligner_lm-20250620T013644Z-1-001.zip
diffusion_transformer-20250620T013642Z-1-001.zip
duration_lm-20250620T013639Z-1-001.zip
g2p-20250620T013633Z-1-001.zip
wavvae-20250620T013627Z-1-001.zip
$ for i in ~/archive/*zip; do unzip $i; done
$ tree -a .
.
├── aligner_lm
│   ├── config.yaml
│   └── model_only_last.ckpt
├── diffusion_transformer
│   ├── config.yaml
│   └── model_only_last.ckpt
├── duration_lm
│   ├── config.yaml
│   └── model_only_last.ckpt
├── g2p
│   ├── added_tokens.json
│   ├── config.json
│   ├── generation_config.json
│   ├── latest
│   ├── merges.txt
│   ├── model.safetensors
│   ├── special_tokens_map.json
│   ├── tokenizer_config.json
│   ├── tokenizer.json
│   ├── trainer_state.json
│   └── vocab.json
└── wavvae
    ├── config.yaml
    └── decoder.ckpt
$ cd ..
$ cp ~/archives/paul.wav assets/.
$ cp ~/archives/book.txt assets/.
$ docker build . -t megatts3:latest
$ docker run -it -p 7929:7929 --gpus all -e CUDA_VISIBLE_DEVICES=0 -v $(pwd)/gen:/app/gen megatts3:latest

Terminal 2:

$ docker exec -it $(docker ps -q | head -n 1) bash
root@hash:/app# python tts/infer_cli.py --input_wav 'assets/paul.wav' --input_text "$(cat book.txt)" --output_dir ./gen --p_w 2.0 --t_w 3.0
Traceback (most recent call last):
  File "/app/tts/infer_cli.py", line 28, in <module>
    from tts.modules.ar_dur.commons.nar_tts_modules import LengthRegulator
ModuleNotFoundError: No module named 'tts'

Recommendations

• Move Model Download section above Requirements (for Docker) in README (download BEFORE build, therefore tell people to download BEFORE build)
• Change # ! You should download the pretrained checkpoint to # ! You MUST download all pretrained checkpoints
• Show how to capture output from docker guest container to host system
• Show how to use text file as input