search

lucidrains / e2-tts-pytorch

Implementation of E2-TTS, "Embarrassingly Easy Fully Non-Autoregressive Zero-Shot TTS", in Pytorch
MIT License
228 stars 21 forks source link

an error in Usage: #2

Closed traddo closed 2 months ago

traddo commented 2 months ago 

Traceback (most recent call last):
  File "/home/steven/work/dub_studio/e2tts.py", line 16, in <module>
    loss = duration_predictor(x, target_duration=duration)
  File "/opt/conda/envs/e2tts/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/opt/conda/envs/e2tts/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
    return forward_call(*args, **kwargs)
TypeError: DurationPredictor.forward() got an unexpected keyword argument 'target_duration'
lucidrains commented 2 months ago

@traddo oops, try removing that kwarg

I'll update the readme later