yi_lu/ti_coding_agent_probe
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add SWIFT coding agent probe experiment scripts

Browse Source
This commit is contained in:
Codex
2026-06-24 21:57:11 +08:00
parent 73e9752331
commit 4063246ac2
3 changed files with 53 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
scripts/setup_env.sh
View File

@@ -9,12 +9,33 @@ export https_proxy="${https_proxy:-http://100.72.0.101:8888}"
export HTTP_PROXY="${HTTP_PROXY:-${http_proxy}}"
export HTTPS_PROXY="${HTTPS_PROXY:-${https_proxy}}"
export PIP_INDEX_URL="${PIP_INDEX_URL:-https://mirrors.aliyun.com/pypi/simple/}"
export UV_DEFAULT_INDEX="${UV_DEFAULT_INDEX:-${PIP_INDEX_URL}}"
export UV_INDEX_URL="${UV_INDEX_URL:-${PIP_INDEX_URL}}"
python3 -m venv .venv
source .venv/bin/activate
python -m pip install -U pip setuptools wheel
python -m pip install -e .
python -m pip install -e third_party/modelscope-swift
python -m pip install -U pip setuptools wheel uv
UV=(python -m uv pip)
"${UV[@]}" install -U pip setuptools wheel
if [[ "${INSTALL_SWIFT_ALL:-0}" == "1" ]]; then
"${UV[@]}" install -e 'third_party/modelscope-swift[all]' --torch-backend=auto
else
"${UV[@]}" install -e third_party/modelscope-swift --torch-backend=auto
fi
"${UV[@]}" install -e .
if [[ "${INSTALL_TRAINING_EXTRAS:-1}" == "1" ]]; then
"${UV[@]}" install -U 'deepspeed<0.19' liger-kernel tensorboard nvitop
fi
if [[ -n "${EXTRA_UV_PACKAGES:-}" ]]; then
# shellcheck disable=SC2206
EXTRA_PACKAGES=(${EXTRA_UV_PACKAGES})
"${UV[@]}" install -U "${EXTRA_PACKAGES[@]}"
fi
mkdir -p data/raw data/processed models outputs runs logs
python - <<'PY'