[update] fix default model conf

WaveGenAI · Jun 24, 2024 · e400964 · e400964
1 parent b14021f
commit e400964
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/inference.py b/scripts/inference.py
@@ -23,7 +23,7 @@
 
 model = SoundStream(
     D=256,
-    C=1,
+    C=58,
     strides=(2, 4, 5, 5),
 )
 

diff --git a/scripts/train.py b/scripts/train.py
@@ -61,7 +61,7 @@
 
 model = SoundStream(
     D=256,
-    C=1,
+    C=58,
     strides=(2, 4, 5, 5),
 )
-Original file line number
+Diff line change
@@ Expand Up / @@ -23,7 +23,7 @@ @@
     model = SoundStream(
         D=256,
-        C=1,
+        C=58,
         strides=(2, 4, 5, 5),
     )
@@ Expand Down @@