From 246c331457edcc0770f7fd3f48cd5c94238d49fd Mon Sep 17 00:00:00 2001 From: "Wu, Zhenyu" Date: Fri, 27 Feb 2026 16:10:02 +0800 Subject: [PATCH] fix: fix wrong dtype --- compressai/models/google.py | 1 + 1 file changed, 1 insertion(+) diff --git a/compressai/models/google.py b/compressai/models/google.py index 7b65ef8e..712ce031 100644 --- a/compressai/models/google.py +++ b/compressai/models/google.py @@ -673,6 +673,7 @@ def decompress(self, strings, shape): y_hat = torch.zeros( (z_hat.size(0), self.M, y_height + 2 * padding, y_width + 2 * padding), device=z_hat.device, + dtype=z_hat.dtype, ) for i, y_string in enumerate(strings[0]):