remove 0.38s infer

dac-vae/inference.py

@@ -137,7 +137,7 @@ class DACVAEInference:
         
         # Forward pass through model
         print("Processing through DACVAE...")
-        audio_tensor = audio_tensor[:, :, :9120]
+        # audio_tensor = audio_tensor[:, :, :9120]
 
         print('audio_tensor shape: ', audio_tensor.shape)
         out = self.model(audio_tensor, self.sample_rate)
@@ -171,7 +171,7 @@ class DACVAEInference:
 
 def main():
     parser = argparse.ArgumentParser(description="DACVAE Audio Inference")
-    parser.add_argument('--checkpoint', type=str, required=False, default="/mnt/nvme/ckpts/24khz/364k_20250702_043748/checkpoint.pt",
+    parser.add_argument('--checkpoint', type=str, required=False, default="checkpoint.pt",
                         help='Path to model checkpoint')
     parser.add_argument('--config', type=str, default="./config.yml",
                         help='Path to config YAML (optional if config is in checkpoint)')