point to specified commit in test case

Signed-off-by: Lize Cai <lize.cai@sap.com>
kserve · Mar 30, 2024 · a592042 · a592042
1 parent b10d356
commit a592042
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/huggingfaceserver/huggingfaceserver/test_model.py b/python/huggingfaceserver/huggingfaceserver/test_model.py
@@ -41,10 +41,12 @@ def test_bert():
 
 
 def test_model_revision():
+    # https://huggingface.co/google-bert/bert-base-uncased
+    commit = "86b5e0934494bd15c9632b12f734a8a67f723594"
     model = HuggingfaceModel("bert-base-uncased",
                              {"model_id": "bert-base-uncased",
-                              "model_revision": "main",
-                              "tokenizer_revision": "main",
+                              "model_revision": commit,
+                              "tokenizer_revision": commit,
                               "disable_lower_case": False}
                              )
     model.load()