From d8111e04770939a2e436865ec98457ccea1ee6c7 Mon Sep 17 00:00:00 2001 From: James Betker Date: Sat, 1 Jan 2022 14:05:33 -0700 Subject: [PATCH] misc --- codes/models/gpt_voice/mini_encoder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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,