Skip to content

Commit

Permalink
Removed duplicate sentence about CUDA hardware that is already mentio…
Browse files Browse the repository at this point in the history
…ned in the intro text. Minor text change.
  • Loading branch information
tstatler committed Sep 4, 2024
1 parent 5fd4f39 commit d5f5adf
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions prototype_source/gpu_quantization_torchao_tutorial.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,15 +35,13 @@
#
# Segment Anything Model checkpoint setup:
#
# 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>``).
# 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>``).
# 2. Pass in that directory by editing the code below to say:
#
# .. code-block:: bash
#
# {sam_checkpoint_base_path}=<path>
#
# This was run on an A100-PG509-200 power limited to 330.00 W
#

import torch
from torchao.quantization import change_linear_weights_to_int8_dqtensors
Expand Down

0 comments on commit d5f5adf

Please sign in to comment.