lucidrains / naturalspeech2-pytorch

Implementation of Natural Speech 2, Zero-shot Speech and Singing Synthesizer, in Pytorch
MIT License
1.29k stars 101 forks source link

Error when following the readme #3

Closed lpy-ET closed 1 year ago

lpy-ET commented 1 year ago

Hello, there seems to be an error in the readme. I tried on both my machine and Colab, and got this error:

File <@beartype(naturalspeech2_pytorch.naturalspeech2_pytorch.NaturalSpeech2.__init__) at 0x7f1ba95b9550>:24, in __init__(__beartype_func, __beartype_conf, __beartype_get_violation, __beartype_object_125973616, __beartype_object_139756782786176, *args, **kwargs)

BeartypeCallHintParamViolation: @beartyped naturalspeech2_pytorch.naturalspeech2_pytorch.NaturalSpeech2.__init__() parameter model="Transformer(
  (layers): ModuleList(
    (0-11): 12 x ModuleList(
      (0): RMSNorm()
     ... violates type hint <class 'naturalspeech2_pytorch.naturalspeech2_pytorch.Model'>, as <class "naturalspeech2_pytorch.naturalspeech2_pytorch.Transformer"> "Transformer(
  (layers): ModuleList(
    (0-11): 12 x ModuleList(
      (0): RMSNorm()
     ... not instance of <class "naturalspeech2_pytorch.naturalspeech2_pytorch.Model">.
lucidrains commented 1 year ago

@lpy-ET oh oops, i forgot to update the readme, thanks!