fix: set LITELLM_LOCAL_MODEL_COST_MAP=True to avoid GitHub fetch timeout
This commit is contained in:
parent
eb0309baad
commit
610d6bb2be
@ -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": [
|
||||||
{
|
{
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user