From 22d6f97d8584dd14ac5d0c5bc4ddad9bf33553fe Mon Sep 17 00:00:00 2001
From: Sayak Paul <spsayakpaul@gmail.com>
Date: Tue, 6 Aug 2024 21:03:17 +0530
Subject: [PATCH] Update serialization.rst to include get_model_size_in_bytes
 import (#604)

---
 docs/source/serialization.rst | 1 +
 1 file changed, 1 insertion(+)

diff --git a/docs/source/serialization.rst b/docs/source/serialization.rst
index c5b597fce0..7cf80aec11 100644
--- a/docs/source/serialization.rst
+++ b/docs/source/serialization.rst
@@ -11,6 +11,7 @@ Here is the serialization and deserialization flow::
   import copy
   import tempfile
   import torch
+  from torchao.utils import get_model_size_in_bytes
   from torchao.quantization.quant_api import (
       quantize_,
       int4_weight_only,