Skip to content

Commit 68ea0ab

Browse files
committed
fix: RasterScanLatentCodec missing None
1 parent 4f3eb62 commit 68ea0ab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

compressai/latent_codecs/rasterscan.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ def __init__(
9595
**kwargs,
9696
):
9797
super().__init__()
98-
self.gaussian_conditional = gaussian_conditional or GaussianConditional()
98+
self.gaussian_conditional = gaussian_conditional or GaussianConditional(None)
9999
self.entropy_parameters = entropy_parameters or nn.Identity()
100100
self.context_prediction = context_prediction or MaskedConv2d()
101101
self.kernel_size = _reduce_seq(self.context_prediction.kernel_size)

0 commit comments

Comments
 (0)