Fix original shape matching with crop (#673)

fishaudio · Nov 10, 2024 · 911e037 · 911e037
1 parent 0e64c4e
commit 911e037
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fish_speech/models/vqgan/modules/fsq.py b/fish_speech/models/vqgan/modules/fsq.py
@@ -99,7 +99,7 @@ def forward(self, z) -> FSQResult:
         if diff > 0:
             result.z = F.pad(result.z, (left, right))
         elif diff < 0:
-            result.z = result.z[..., left:-right]
+            result.z = result.z[..., -left:right]
 
         return result