fix: increasing H200 test timeout (#15600)

This commit is contained in:
Douglas Yang
2025-12-23 01:00:37 -08:00
committed by GitHub
parent 66902e0f1b
commit f9dd90ac35
5 changed files with 27 additions and 8 deletions

View File

@@ -168,6 +168,11 @@ def is_blackwell_system():
return envs.IS_BLACKWELL.get()
def is_h200_system():
"""Return whether it is running on an H200 system."""
return envs.IS_H200.get()
def _use_cached_default_models(model_repo: str):
cache_dir = os.getenv("DEFAULT_MODEL_CACHE_DIR")
if cache_dir and model_repo:
@@ -193,6 +198,9 @@ if is_in_amd_ci():
if is_blackwell_system():
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH = 3000
if is_h200_system():
DEFAULT_TIMEOUT_FOR_SERVER_LAUNCH = 3600
def call_generate_lightllm(prompt, temperature, max_tokens, stop=None, url=None):
assert url is not None