fix: set LITELLM_LOCAL_MODEL_COST_MAP=True to avoid GitHub fetch timeout

This commit is contained in:
sora 2026-08-03 07:14:05 +00:00
parent eb0309baad
commit 610d6bb2be
2 changed files with 10 additions and 3 deletions

View File

@ -31,6 +31,9 @@ import subprocess
import sys import sys
from pathlib import Path from pathlib import Path
# 避免 LiteLLM 联网拉取 model cost map 导致超时
os.environ.setdefault('LITELLM_LOCAL_MODEL_COST_MAP', 'True')
def _find_repo_root() -> Path: def _find_repo_root() -> Path:
"""向上查找,直到目录下存在 bash/run.py。""" """向上查找,直到目录下存在 bash/run.py。"""
p = Path(__file__).resolve().parent p = Path(__file__).resolve().parent
@ -44,9 +47,8 @@ def _find_repo_root() -> Path:
ROOT = _find_repo_root() ROOT = _find_repo_root()
# --------------------------------------------------------------------------- deep_swe,
# Kimi K3 report -> evalscope dataset 映射
# ---------------------------------------------------------------------------
BENCHMARKS = { BENCHMARKS = {
"Coding": [ "Coding": [
{ {

View File

@ -30,12 +30,17 @@ Examples:
import argparse import argparse
import json import json
import os
import sys import sys
import threading import threading
import time import time
from copy import deepcopy from copy import deepcopy
from pathlib import Path from pathlib import Path
# LiteLLM 默认会联网拉取 GitHub 上的 model cost map内网/无网环境会超时。
# 强制使用本地备份,避免运行时被卡住。
os.environ.setdefault('LITELLM_LOCAL_MODEL_COST_MAP', 'True')
import yaml import yaml
from evalscope import run_task, TaskConfig from evalscope import run_task, TaskConfig