Skip to content

Commit

Permalink
Update serialization.rst to include get_model_size_in_bytes import (#604
Browse files Browse the repository at this point in the history
)
  • Loading branch information
sayakpaul authored Aug 6, 2024
1 parent 1b1e94c commit 22d6f97
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions docs/source/serialization.rst
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 22d6f97

Please sign in to comment.