Error during audio generation or saving: '<' not supported between instances of 'int' and 'NoneType'

[quocvu]

I am using python 3.12 and got this error:

Running command: cd /root/dia_repo && python cli.py --local-paths --config /data_mnt/models/nari-labs/Dia-1.6B-0626/config.json --checkpoint /data_mnt/models/nari-labs/Dia-1.6B-0626/dia-v1.pth --audio-prompt /data_mnt/voices/short/british_man_brian_aws.wav --output /tmp/tmpylwxubom.mp3 "The quick brown fox jumps over the lazy dog. Every morning, she walks five miles to the old bakery."
/usr/local/lib/python3.12/site-packages/torch/nn/utils/weight_norm.py:143: FutureWarning: `torch.nn.utils.weight_norm` is deprecated in favor of `torch.nn.utils.parametrizations.weight_norm`.
  WeightNorm.apply(module, name, dim)
Using device: cuda
Loading model...
Loading from local paths: config='/data_mnt/models/nari-labs/Dia-1.6B-0626/config.json', checkpoint='/data_mnt/models/nari-labs/Dia-1.6B-0626/dia-v1.pth'
Model loaded.
Generating audio...
Error during audio generation or saving: '<' not supported between instances of 'int' and 'NoneType'