diff --git a/gsplat/compression/png_compression.py b/gsplat/compression/png_compression.py index 2dbece8cf..ee4ead6f9 100644 --- a/gsplat/compression/png_compression.py +++ b/gsplat/compression/png_compression.py @@ -368,7 +368,9 @@ def _compress_kmeans( maxs = torch.max(centroids) centroids_norm = (centroids - mins) / (maxs - mins) centroids_norm = centroids_norm.detach().cpu().numpy() - centroids_quant = (centroids_norm * (2**quantization - 1)).round().astype(np.uint8) + centroids_quant = ( + (centroids_norm * (2**quantization - 1)).round().astype(np.uint8) + ) labels = labels.astype(np.uint16) npz_dict = {