From 883a450bf57bd6825a1f40bc82e8a266398db67e Mon Sep 17 00:00:00 2001 From: Lize Cai Date: Sat, 30 Mar 2024 11:04:13 +0800 Subject: [PATCH] point to specified commit in test case Signed-off-by: Lize Cai --- python/huggingfaceserver/huggingfaceserver/test_model.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/huggingfaceserver/huggingfaceserver/test_model.py b/python/huggingfaceserver/huggingfaceserver/test_model.py index 3b7e3a4dc10..8a954171355 100644 --- a/python/huggingfaceserver/huggingfaceserver/test_model.py +++ b/python/huggingfaceserver/huggingfaceserver/test_model.py @@ -41,10 +41,11 @@ def test_bert(): def test_model_revision(): + # https://huggingface.co/google-bert/bert-base-uncased model = HuggingfaceModel("bert-base-uncased", {"model_id": "bert-base-uncased", - "model_revision": "main", - "tokenizer_revision": "main", + "model_revision": "86b5e0934494bd15c9632b12f734a8a67f723594", + "tokenizer_revision": "86b5e0934494bd15c9632b12f734a8a67f723594", "disable_lower_case": False} ) model.load()