Skip to content

Commit 0ba5732

Browse files
authored
Update model ckpt download url in prepare_model.py (#1716)
1 parent ba75648 commit 0ba5732

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/onnxrt/image_recognition/beit/quantization/ptq_static/prepare_model.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from timm.models import create_model
77
import beit_modeling_finetune
88

9-
MODEL_URLS = {"beit_base_patch16_224": "https://conversationhub.blob.core.windows.net/beit-share-public/beit/beit_base_patch16_224_pt22k_ft22kto1k.pth?sv=2021-10-04&st=2023-06-08T11%3A16%3A02Z&se=2033-06-09T11%3A16%3A00Z&sr=c&sp=r&sig=N4pfCVmSeq4L4tS8QbrFVsX6f6q844eft8xSuXdxU48%3D",}
9+
MODEL_URLS = {"beit_base_patch16_224": "https://github.com/addf400/files/releases/download/v1.0/beit_base_patch16_224_pt22k_ft22kto1k.pth",}
1010
MODEL_FILES = {"beit_base_patch16_224": "beit_base_patch16_224_pt22k_ft22kto1k.pth"}
1111
MAX_TIMES_RETRY_DOWNLOAD = 5
1212

@@ -99,4 +99,4 @@ def prepare_model(input_model, output_model):
9999

100100
if __name__ == "__main__":
101101
args = parse_arguments()
102-
prepare_model(args.input_model, args.output_model)
102+
prepare_model(args.input_model, args.output_model)

0 commit comments

Comments
 (0)