diff --git a/codes/models/gpt_voice/mini_encoder.py b/codes/models/gpt_voice/mini_encoder.py index 8dd99883..bc04b1a5 100644 --- a/codes/models/gpt_voice/mini_encoder.py +++ b/codes/models/gpt_voice/mini_encoder.py @@ -90,7 +90,8 @@ class ResBlock(nn.Module): class AudioMiniEncoder(nn.Module): - def __init__(self, spec_dim, + def __init__(self, + spec_dim, embedding_dim, base_channels=128, depth=2,