Merge origin/main into dev/vllm-align; resolve omni_stage.py conflicts
Co-authored-by: Cursor <cursoragent@cursor.com> Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
This commit is contained in:
commit
03c6a58f69
18 files changed
+761
-381
No files matched your search
+4
-3
@@ -197,12 +197,13 @@ class Qwen3TTSTokenizerV2CausalTransConvNet(nn.Module):
|
||||
self.conv = nn.ConvTranspose1d(in_channels, out_channels, kernel_size, stride=stride)
|
||||
|
||||
pad = kernel_size - stride
|
||||
self.left_pad = math.ceil(pad)
|
||||
self.right_pad = pad = self.left_pad
|
||||
self.left_pad = 0
|
||||
self.right_pad = int(pad)
|
||||
|
||||
def forward(self, hidden_state):
|
||||
hidden_state = self.conv(hidden_state)
|
||||
hidden_state = hidden_state[..., self.left_pad : hidden_state.shape[-1] - self.right_pad]
|
||||
if self.right_pad > 0:
|
||||
hidden_state = hidden_state[..., : hidden_state.shape[-1] - self.right_pad]
|
||||
return hidden_state.contiguous()
|
||||
|
||||
|
||||
|
||||
Reference in new issue
Block a user