update: add UniDriveVLA deps into prepare_env.sh

prepare_env.sh

@@ -95,7 +95,7 @@ echo "[4/7] Checking model & dataset libraries ..."
 
 pip3 install --quiet \
     'timm>=1.0' \
-    transformers \
+    'transformers>=5.0' \
     datasets \
     torchvision \
     accelerate \
@@ -112,11 +112,12 @@ echo "  datasets:      $(python3 -c 'import datasets; print(datasets.__version__
 echo ""
 echo "[5/7] Checking NVIDIA ModelOpt ..."
 
-if python3 -c "import modelopt.onnx.quantization" 2>/dev/null; then
+if python3 -c "import modelopt.onnx.quantization" 2>/dev/null && \
+   python3 -c "import modelopt; assert modelopt.__version__ >= '0.44.0'" 2>/dev/null; then
     echo "  modelopt already installed: $(pip3 show nvidia-modelopt 2>/dev/null | grep Version)"
 else
-    echo "  Installing nvidia-modelopt ..."
-    pip3 install --quiet nvidia-modelopt
+    echo "  Installing nvidia-modelopt==0.44.0 ..."
+    pip3 install --quiet 'nvidia-modelopt==0.44.0'
 fi
 
 # Also install nvidia-ml-py (dependency for TensorRT / modelopt)
@@ -151,6 +152,32 @@ echo "  numpy:        $(python3 -c 'import numpy; print(numpy.__version__)')"
 echo "  scikit-learn: $(python3 -c 'import sklearn; print(sklearn.__version__)')"
 echo "  paramiko:     $(python3 -c 'import paramiko; print(paramiko.__version__)')"
 
+# ---------------------------------------------------------------------------
+# 6b. UniDriveVLA / DriveLM evaluation dependencies
+# ---------------------------------------------------------------------------
+# The unidrive_vla_nusc_base_evaluation.py script computes DriveLM metrics that
+# the original prepare_env.sh did not cover:
+#   - nltk, rouge-score, pycocoevalcap : BLEU/ROUGE-L/CIDEr (tag 2 language score)
+#   - sentence-transformers            : semantic-similarity fallback for GPT-Score
+#   - qwen-vl-utils                   : multi-view image processing for Qwen3-VL
+# Each is installed idempotently (pip skips if already satisfied). These pull
+# from the configured pip mirror (Tencent cloud), not huggingface.co.
+echo ""
+echo "[6b/7] Checking UniDriveVLA / DriveLM evaluation dependencies ..."
+
+pip3 install --quiet \
+    nltk \
+    rouge-score \
+    pycocoevalcap \
+    sentence-transformers \
+    qwen-vl-utils
+
+echo "  nltk:                  $(python3 -c 'import nltk; print(nltk.__version__)')"
+echo "  rouge_score:           $(python3 -c 'import rouge_score; print("OK")' 2>&1 | tail -1)"
+echo "  pycocoevalcap:         $(python3 -c 'import pycocoevalcap; print("OK")' 2>&1 | tail -1)"
+echo "  sentence-transformers: $(python3 -c 'import sentence_transformers; print(sentence_transformers.__version__)')"
+echo "  qwen-vl-utils:         $(python3 -c 'import qwen_vl_utils; print("OK")' 2>&1 | tail -1)"
+
 # ---------------------------------------------------------------------------
 # 7. Offline environment for HuggingFace (no internet access to HF)
 # ---------------------------------------------------------------------------