Bump Huggingface timeout from 15 to 20 minutes. (#627)

mlcommons · Oct 22, 2024 · 1d9b8b0 · 1d9b8b0
1 parent 286f892
commit 1d9b8b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/huggingface/modelgauge/suts/huggingface_chat_completion.py b/plugins/huggingface/modelgauge/suts/huggingface_chat_completion.py
@@ -17,7 +17,7 @@
 from modelgauge.sut_decorator import modelgauge_sut
 from modelgauge.sut_registry import SUTS
 
-HUGGING_FACE_TIMEOUT = 60 * 15
+HUGGING_FACE_TIMEOUT = 60 * 20
 
 
 class ChatMessage(BaseModel):