import torch from naturalspeech2_pytorch import ( EncodecWrapper, Model, NaturalSpeech2, SpeechPromptEncoder ) # use encodec as an example codec = EncodecWrapper() model = Model( dim = 128, depth = 6, dim_prompt = 512, cond_drop_prob = 0.25, # dropout prompt conditioning with this pr...