point to specified commit in test case

Signed-off-by: Lize Cai <lize.cai@sap.com>
kserve · Mar 30, 2024 · 883a450 · 883a450
1 parent b10d356
commit 883a450
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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()