The package fails to reference naturalspeech2_pytorch.utils.*

[ArEnSc]

^
on an related @lucidrains is this ready for testing? I am thinking about trying to get it to memorize a sample to test and then pretrain a model. I did notice that you mentioned that the pitch conditioning wasn't complete

[lucidrains]

@ArEnSc hmm, can you show me the full stack trace?

[ArEnSc]

@lucidrains

@ArEnSc hmm, can you show me the full stack trace?

│ /usr/local/lib/python3.10/dist-packages/IPython/core/interactiveshell.py:3553 in run_code │
│ │
│ 3550 │ │ │ │ elif async_ : │
│ 3551 │ │ │ │ │ await eval(code_obj, self.user_global_ns, self.user_ns) │
│ 3552 │ │ │ │ else: │
│ ❱ 3553 │ │ │ │ │ exec(code_obj, self.user_global_ns, self.user_ns) │
│ 3554 │ │ │ finally: │
│ 3555 │ │ │ │ # Reset our crash handler in place │
│ 3556 │ │ │ │ sys.excepthook = old_excepthook │
│ in <cell line: 2>:2 │
│ │
│ /usr/local/lib/python3.10/dist-packages/naturalspeech2_pytorch/init.py:8 in │
│ │
│ 5 │ from einops._torch_specific import allow_ops_in_compiled_graph │
│ 6 │ allow_ops_in_compiled_graph() │
│ 7 │
│ ❱ 8 from naturalspeech2_pytorch.naturalspeech2_pytorch import ( │
│ 9 │ NaturalSpeech2, │
│ 10 │ Transformer, │
│ 11 │ Wavenet, │
│ │
│ /usr/local/lib/python3.10/dist-packages/naturalspeech2_pytorch/naturalspeech2_pytorch.py:27 in │
│ │
│ │
│ 24 from beartype.door import is_bearable │
│ 25 │
│ 26 from naturalspeech2_pytorch.attend import Attend │
│ ❱ 27 from naturalspeech2_pytorch.utils.tokenizer import Tokenizer, ESpeak │
│ 28 from naturalspeech2_pytorch.version import version │
│ 29 │
│ 30 from accelerate import Accelerator │
╰──────────────────────────────────────────────────────────────────────────────────────────────────╯
ModuleNotFoundError: No module named 'naturalspeech2_pytorch.utils.tokenizer'

reproduce
!pip install naturalspeech2-pytorch
then paste in

import torch
from naturalspeech2_pytorch import (
EncodecWrapper,
Model,
NaturalSpeech2
)

[lucidrains]

@ArEnSc would you like to retry with 0.0.38?

[ArEnSc]

@lucidrains it works!
Quick question so is it ready for training minus the pitch conditioning ?