| # Shared interpreter for every script here. Source, do not execute. | |
| # | |
| # Inference runs in the Spatial_VLM conda env (transformers 5.7, torch 2.8). | |
| # Qwen3.5 (model_type qwen3_5) does not exist in the base env's transformers | |
| # 4.51, so a silent fallback to `python` would let the Qwen2.5 sweeps pass and | |
| # then fail halfway through the grid. The interpreter is therefore required, | |
| # not preferred: a missing one is an error here rather than a confusing failure | |
| # four hours in. Override deliberately with PYTHON_BIN=... | |
| SPATIAL_VLM_PYTHON=${SPATIAL_VLM_PYTHON:-/home/disheng/miniconda3/envs/Spatial_VLM/bin/python} | |
| PYTHON_BIN=${PYTHON_BIN:-${SPATIAL_VLM_PYTHON}} | |
| if [[ ! -x "${PYTHON_BIN}" ]]; then | |
| echo "ERROR: interpreter not found or not executable: ${PYTHON_BIN}" >&2 | |
| echo " Inference needs the Spatial_VLM env (transformers 5.7, torch 2.8)." >&2 | |
| echo " Set PYTHON_BIN=/path/to/python to use a different one." >&2 | |
| exit 1 | |
| fi | |
| # Confirm the env can actually serve this grid. Costs one interpreter start per | |
| # sweep, which is nothing against the run, and turns a wrong env into a clear | |
| # message up front. CHECK_ENV=false to skip. | |
| check_env() { | |
| [[ "${CHECK_ENV:-true}" == "true" ]] || return 0 | |
| "${PYTHON_BIN}" - <<'PY' || exit 1 | |
| import sys | |
| try: | |
| import torch, transformers | |
| except Exception as exc: # noqa: BLE001 | |
| sys.exit(f"ERROR: cannot import torch/transformers: {exc}") | |
| # qwen3_5 is the model_type behind Qwen3.5-4B and -4B-Base; it does not | |
| # exist before transformers 5.x. | |
| try: | |
| from transformers.models import qwen3_5 # noqa: F401 | |
| except ImportError: | |
| sys.exit( | |
| f"ERROR: transformers {transformers.__version__} at {sys.executable}\n" | |
| " has no qwen3_5 support, so the Qwen3.5 sweeps cannot run.\n" | |
| " Use the Spatial_VLM env (transformers 5.7)." | |
| ) | |
| print(f"env ok: python {sys.version.split()[0]}, torch {torch.__version__}, " | |
| f"transformers {transformers.__version__}, " | |
| f"cuda {'yes' if torch.cuda.is_available() else 'NO'}") | |
| PY | |
| } | |