diff --git a/compressai/entropy_models/entropy_models.py b/compressai/entropy_models/entropy_models.py index 6c309cdb..c3ccb4d9 100644 --- a/compressai/entropy_models/entropy_models.py +++ b/compressai/entropy_models/entropy_models.py @@ -677,7 +677,6 @@ def update(self): tail_mass = 2 * lower[:, :1] - quantized_cdf = torch.Tensor(len(pmf_length), max_length + 2) quantized_cdf = self._pmf_to_cdf(pmf, tail_mass, pmf_length, max_length) self._quantized_cdf = quantized_cdf self._offset = -pmf_center