misc

2022-01-01 14:05:33 -07:00 · 2022-01-01 14:05:33 -07:00 · d8111e0477
commit d8111e0477
parent dc535b5358
1 changed files with 2 additions and 1 deletions
--- 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,