Pārlūkot izejas kodu

Fix original shape matching with crop (#673)

Yuchao Zhang 1 gadu atpakaļ
vecāks
revīzija
911e037855
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      fish_speech/models/vqgan/modules/fsq.py

+ 1 - 1
fish_speech/models/vqgan/modules/fsq.py

@@ -99,7 +99,7 @@ class DownsampleFiniteScalarQuantize(nn.Module):
         if diff > 0:
         if diff > 0:
             result.z = F.pad(result.z, (left, right))
             result.z = F.pad(result.z, (left, right))
         elif diff < 0:
         elif diff < 0:
-            result.z = result.z[..., left:-right]
+            result.z = result.z[..., -left:right]
 
 
         return result
         return result