Merge pull request #15 from piwell/fix-pretrained-dac-pretransform

fix: call model.quantizer with latents in PretrainedDACPretransform

Author: zqevans

stable_audio_tools/models/pretransforms.py

@@ -106,7 +106,7 @@ def encode(self, x):
         if self.quantize_on_decode:
             output = latents
         else:
-            z, _, latents, _, _ = self.model.quantizer(z, n_quantizers=self.model.n_codebooks)
+            z, _, _, _, _ = self.model.quantizer(latents, n_quantizers=self.model.n_codebooks)
             output = z
 
         if self.scale != 1.0: