Skip to content

Commit e16b8e8

Browse files
committed
Don't unzip, download to the prototype/data dir
1 parent cbdd8b7 commit e16b8e8

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

.jenkins/download_data.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -108,11 +108,9 @@ def download_lenet_mnist() -> None:
108108
def download_gpu_quantization_torchao() -> None:
109109
# Download SAM model checkpoint for prototype_source/gpu_quantization_torchao_tutorial.py
110110
z = download_url_to_file("https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
111-
prefix=DATA_DIR,
111+
prefix=PROTOTYPE_DATA_DIR,
112112
sha256="a7bf3b02f3ebf1267aba913ff637d9a2d5c33d3173bb679e46d9f338c26f262e",
113113
)
114-
unzip(z, PROTOTYPE_DATA_DIR)
115-
116114

117115
def main() -> None:
118116
DATA_DIR.mkdir(exist_ok=True)

0 commit comments

Comments
 (0)