fix: correct transpose conv context
This commit is contained in:
@@ -537,7 +537,7 @@ class StreamingVAEDecoder:
|
|||||||
self._patch_causal_conv(mod, pad)
|
self._patch_causal_conv(mod, pad)
|
||||||
elif isinstance(mod, CausalTransposeConv1d):
|
elif isinstance(mod, CausalTransposeConv1d):
|
||||||
trim = mod._CausalTransposeConv1d__padding * 2 - mod._CausalTransposeConv1d__output_padding
|
trim = mod._CausalTransposeConv1d__padding * 2 - mod._CausalTransposeConv1d__output_padding
|
||||||
ctx = mod.kernel_size[0] // mod.stride[0] - 1
|
ctx = (mod.kernel_size[0] - 1) // mod.stride[0]
|
||||||
if ctx > 0:
|
if ctx > 0:
|
||||||
self._patch_transpose_conv(mod, ctx, trim)
|
self._patch_transpose_conv(mod, ctx, trim)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user