From 1a46c5d1ad724097200d1b3fa713e1baad595cf8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=91=A8=E9=80=B8=E8=BD=A9?=
 <modelbest@zhouyixuandeMacBook-Pro.local>
Date: Thu, 18 Sep 2025 14:53:37 +0800
Subject: [PATCH] update README

---
 src/voxcpm/model/voxcpm.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/voxcpm/model/voxcpm.py b/src/voxcpm/model/voxcpm.py
index 0d8f1c2..df13188 100644
--- a/src/voxcpm/model/voxcpm.py
+++ b/src/voxcpm/model/voxcpm.py
@@ -283,10 +283,11 @@ class VoxCPMModel(nn.Module):
                 else:
                     break
             else:
+                break   
                 
         decode_audio = self.audio_vae.decode(latent_pred.to(torch.float32)).squeeze(1).cpu()  
         decode_audio = decode_audio[..., 640:-640] # trick: trim the start and end of the audio
-        return decode_audio
+        return decode_audio        
     
     @torch.inference_mode()
     def build_prompt_cache(