From 14ca7f1452d8fd91f9459a4a069bd3b9455ea216 Mon Sep 17 00:00:00 2001 From: Fanli Lin Date: Sat, 8 Feb 2025 04:42:09 +0800 Subject: [PATCH] [docs] fix typo (#36080) typo fix --- docs/source/en/quantization/compressed_tensors.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/quantization/compressed_tensors.md b/docs/source/en/quantization/compressed_tensors.md index 77b8986bf8..177e261445 100644 --- a/docs/source/en/quantization/compressed_tensors.md +++ b/docs/source/en/quantization/compressed_tensors.md @@ -61,7 +61,7 @@ ct_model = AutoModelForCausalLM.from_pretrained("nm-testing/Meta-Llama-3.1-8B-In # Measure memory usage mem_params = sum([param.nelement()*param.element_size() for param in ct_model.parameters()]) -print(f"{mem/2**30:.4f} GB") +print(f"{mem_params/2**30:.4f} GB") # 8.4575 GB ```