[docs] fix typo (#36080)

typo fix
This commit is contained in:
Fanli Lin
2025-02-08 04:42:09 +08:00
committed by GitHub
parent c361b1e3d9
commit 14ca7f1452

View File

@@ -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
```