Skip to content

Commit d5f5adf

Browse files
committed
Removed duplicate sentence about CUDA hardware that is already mentioned in the intro text. Minor text change.
1 parent 5fd4f39 commit d5f5adf

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

prototype_source/gpu_quantization_torchao_tutorial.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,15 +35,13 @@
3535
#
3636
# Segment Anything Model checkpoint setup:
3737
#
38-
# 1. Go to the `segment-anything repo checkpoint <https://github.com/facebookresearch/segment-anything/tree/main#model-checkpoints>`_ and download the ``vit_h`` checkpoint. Alternatively, you can use ``wget`` (for example: ``wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth --directory-prefix=<path>``).
38+
# 1. Go to the `segment-anything repo checkpoint <https://github.com/facebookresearch/segment-anything/tree/main#model-checkpoints>`_ and download the ``vit_h`` checkpoint. Alternatively, you can use ``wget`` (for example, ``wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth --directory-prefix=<path>``).
3939
# 2. Pass in that directory by editing the code below to say:
4040
#
4141
# .. code-block:: bash
4242
#
4343
# {sam_checkpoint_base_path}=<path>
4444
#
45-
# This was run on an A100-PG509-200 power limited to 330.00 W
46-
#
4745

4846
import torch
4947
from torchao.quantization import change_linear_weights_to_int8_dqtensors

0 commit comments

Comments
 (0)