launch_env.sh

#!/bin/bash

# Script: launch_env.sh
# Purpose: Activate ML environment with all optimizations
# Usage: source launch_env.sh [--auto] [ENV_NAME|1-156]

WORKSPACE_DIR="$(cd -- "$(dirname -- "${BASH_SOURCE[0]}")/.." && pwd -P)"

# Source bashrc to ensure environment is properly loaded
if [ -f "$HOME/.bashrc" ]; then
    source "$HOME/.bashrc"
fi

# Colors
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m'
print_info() { echo -e "${GREEN}[INFO]${NC} $1"; }
print_error() { echo -e "${RED}[ERROR]${NC} $1"; }
print_warning() { echo -e "${YELLOW}[WARNING]${NC} $1"; }

prepend_env_path_once() {
    local var_name="$1"
    local dir="$2"
    local current="${!var_name:-}"

    [ -n "$dir" ] || return 0
    [ -d "$dir" ] || return 0
    case ":$current:" in
        *":$dir:"*) return 0 ;;
    esac

    if [ -n "$current" ]; then
        export "$var_name=$dir:$current"
    else
        export "$var_name=$dir"
    fi
}

cuda_home_is_valid() {
    local cuda_home="$1"

    [ -n "$cuda_home" ] || return 1
    cuda_home="${cuda_home%/}"
    [ -d "$cuda_home" ] || return 1
    [ -x "$cuda_home/bin/nvcc" ] || return 1
}

cuda_version_for_home() {
    local cuda_home="$1"

    cuda_home="${cuda_home%/}"
    if ! cuda_home_is_valid "$cuda_home"; then
        echo ""
        return 1
    fi

    "$cuda_home/bin/nvcc" --version 2>/dev/null | sed -n 's/.*release \([0-9][0-9]*\.[0-9][0-9]*\).*/\1/p' | head -1
}

detect_default_cuda_home() {
    local candidates=()
    local nvcc_path=""

    candidates+=("/usr/local/cuda")
    if [ -n "${CUDA_HOME:-}" ]; then
        candidates+=("$CUDA_HOME")
    fi
    if [ -n "${CUDA_PATH:-}" ]; then
        candidates+=("$CUDA_PATH")
    fi
    if nvcc_path=$(command -v nvcc 2>/dev/null); then
        candidates+=("$(cd -- "$(dirname -- "$nvcc_path")/.." && pwd)")
    fi

    local candidate
    local seen=":"
    for candidate in "${candidates[@]}"; do
        [ -n "$candidate" ] || continue
        candidate="${candidate%/}"
        case "$seen" in
            *":$candidate:"*) continue ;;
        esac
        seen+="$candidate:"

        if cuda_home_is_valid "$candidate"; then
            echo "$candidate"
            return 0
        fi
    done

    return 1
}

apply_env_cuda_selection() {
    local cuda_config="$ENV_PATH/.cuda_env"
    CUDA_ENV_MODE="bashrc"
    CUDA_ENV_HOME=""

    if [ -f "$cuda_config" ]; then
        source "$cuda_config"
    fi

    case "${CUDA_ENV_MODE:-bashrc}" in
        explicit)
            if ! cuda_home_is_valid "$CUDA_ENV_HOME"; then
                print_error "Selected CUDA toolkit is not available: $CUDA_ENV_HOME"
                print_error "Install CUDA first with ./installers/02_install_cuda.sh or rerun ./installers/05_setup_env.sh to select another CUDA version."
                return 1
            fi
            export CUDA_HOME="${CUDA_ENV_HOME%/}"
            export CUDA_PATH="$CUDA_HOME"
            export ML_ENV_CUDA_SOURCE="environment selection"
            ;;
        bashrc|"")
            local default_cuda_home
            if ! default_cuda_home=$(detect_default_cuda_home); then
                print_error "No CUDA toolkit detected. Install CUDA first with ./installers/02_install_cuda.sh."
                return 1
            fi
            export CUDA_HOME="${default_cuda_home%/}"
            export CUDA_PATH="$CUDA_HOME"
            export CUDA_ENV_MODE="bashrc"
            export ML_ENV_CUDA_SOURCE="bashrc default"
            ;;
        *)
            print_error "Invalid CUDA_ENV_MODE in $cuda_config: $CUDA_ENV_MODE"
            return 1
            ;;
    esac

    prepend_env_path_once PATH "$CUDA_HOME/bin"
    prepend_env_path_once LD_LIBRARY_PATH "$CUDA_HOME/lib64"
    prepend_env_path_once LD_LIBRARY_PATH "$CUDA_HOME/lib"
    prepend_env_path_once LIBRARY_PATH "$CUDA_HOME/lib64"
    prepend_env_path_once LIBRARY_PATH "$CUDA_HOME/lib"
    if [ -d "$CUDA_HOME/include/cccl" ]; then
        prepend_env_path_once CPATH "$CUDA_HOME/include/cccl"
    fi

    export ML_ENV_CUDA_APPLIED=1
    export ML_ENV_CUDA_HOME="$CUDA_HOME"
    export ML_ENV_CUDA_VERSION
    ML_ENV_CUDA_VERSION=$(cuda_version_for_home "$CUDA_HOME")
    print_info "CUDA toolkit: $ML_ENV_CUDA_HOME (${ML_ENV_CUDA_VERSION:-unknown}, $ML_ENV_CUDA_SOURCE)"
}

resolve_env_type() {
    local input="${1#env_}"

    case "$input" in
        1|h200_allenai_vllm|h200-allenai-vllm)
            echo "h200-allenai-vllm"
            ;;
        2|h200_arcee_nvfp4_vllm|h200-arcee-nvfp4-vllm)
            echo "h200-arcee-nvfp4-vllm"
            ;;
        3|h200_arcee_vllm|h200-arcee-vllm)
            echo "h200-arcee-vllm"
            ;;
        4|h200_arcee_vllm_pr_54479|h200-arcee-vllm-pr-54479)
            echo "h200-arcee-vllm-pr-54479"
            ;;
        5|h200_arcee_vllm_pr_54479_fp8_block|h200-arcee-vllm-pr-54479-fp8-block)
            echo "h200-arcee-vllm-pr-54479-fp8-block"
            ;;
        6|h200_arcee_vllm_pr_54479_thinking_fp8_block|h200-arcee-vllm-pr-54479-thinking-fp8-block)
            echo "h200-arcee-vllm-pr-54479-thinking-fp8-block"
            ;;
        7|h200_cohere_vllm|h200-cohere-vllm)
            echo "h200-cohere-vllm"
            ;;
        8|h200_cohere_vllm_pr_54479|h200-cohere-vllm-pr-54479)
            echo "h200-cohere-vllm-pr-54479"
            ;;
        9|h200_datalab_vllm|h200-datalab-vllm)
            echo "h200-datalab-vllm"
            ;;
        10|h200_deepseek_sglang|h200-deepseek-sglang)
            echo "h200-deepseek-sglang"
            ;;
        11|h200_deepseek_v41_1017_vllm_5df63a720|h200-deepseek-v41-1017-vllm-5df63a720)
            echo "h200-deepseek-v41-1017-vllm-5df63a720"
            ;;
        12|h200_deepseek_v41_vllm_e77daef89|h200-deepseek-v41-vllm-e77daef89)
            echo "h200-deepseek-v41-vllm-e77daef89"
            ;;
        13|h200_deepseek_vision_sglang_pr_37253|h200-deepseek-vision-sglang-pr-37253)
            echo "h200-deepseek-vision-sglang-pr-37253"
            ;;
        14|h200_deepseek_vision_vllm_pr_54566|h200-deepseek-vision-vllm-pr-54566)
            echo "h200-deepseek-vision-vllm-pr-54566"
            ;;
        15|h200_deepseek_vllm|h200-deepseek-vllm)
            echo "h200-deepseek-vllm"
            ;;
        16|h200_diffusiongemma_sglang|h200-diffusiongemma-sglang)
            echo "h200-diffusiongemma-sglang"
            ;;
        17|h200_gemma_sglang|h200-gemma-sglang)
            echo "h200-gemma-sglang"
            ;;
        18|h200_gemma_vllm|h200-gemma-vllm|h200_gemma4_vllm|h200-gemma4-vllm|h200_gemma_4_vllm|h200-gemma-4-vllm)
            echo "h200-gemma-vllm"
            ;;
        19|h200_gemma3n_vllm|h200-gemma3n-vllm)
            echo "h200-gemma3n-vllm"
            ;;
        20|h200_glm53_vllm_v0290|h200-glm53-vllm-v0290)
            echo "h200-glm53-vllm-v0290"
            ;;
        21|h200_glm53flash_dflash2_sglang_pr_37818|h200-glm53flash-dflash2-sglang-pr-37818)
            echo "h200-glm53flash-dflash2-sglang-pr-37818"
            ;;
        22|h200_glm53flash_dflash2_vllm_pr_55423|h200-glm53flash-dflash2-vllm-pr-55423)
            echo "h200-glm53flash-dflash2-vllm-pr-55423"
            ;;
        23|h200_glm53flash_vllm_pr_53906|h200-glm53flash-vllm-pr-53906)
            echo "h200-glm53flash-vllm-pr-53906"
            ;;
        24|h200_gpt_oss_sglang|h200_gptoss_sglang|h200_gpt-oss_sglang|h200-gptoss-sglang|h200-gpt-oss-sglang)
            echo "h200-gpt-oss-sglang"
            ;;
        25|h200_gpt_oss_vllm|h200_gptoss_vllm|h200-gpt-oss_vllm|h200_vllm_gptoss|h200-gptoss-vllm|h200-gpt-oss-vllm)
            echo "h200-gpt-oss-vllm"
            ;;
        26|h200_ibm_sglang|h200-ibm-sglang)
            echo "h200-ibm-sglang"
            ;;
        27|h200_ibm_vllm|h200-ibm-vllm)
            echo "h200-ibm-vllm"
            ;;
        28|h200_inclusionai_ling3_vllm|h200-inclusionai-ling3-vllm)
            echo "h200-inclusionai-ling3-vllm"
            ;;
        29|h200_inclusionai_sglang|h200-inclusionai-sglang)
            echo "h200-inclusionai-sglang"
            ;;
        30|h200_inclusionai_vllm|h200-inclusionai-vllm)
            echo "h200-inclusionai-vllm"
            ;;
        31|h200_incoai_sglang|h200-incoai-sglang)
            echo "h200-incoai-sglang"
            ;;
        32|h200_incoai_vllm|h200-incoai-vllm)
            echo "h200-incoai-vllm"
            ;;
        33|h200_intel_sglang|h200-intel-sglang)
            echo "h200-intel-sglang"
            ;;
        34|h200_intel_vllm|h200-intel-vllm)
            echo "h200-intel-vllm"
            ;;
        35|h200_liquidai_sglang|h200-liquidai-sglang)
            echo "h200-liquidai-sglang"
            ;;
        36|h200_liquidai_sglang_pr_31041|h200-liquidai-sglang-pr-31041)
            echo "h200-liquidai-sglang-pr-31041"
            ;;
        37|h200_liquidai_vllm|h200-liquidai-vllm)
            echo "h200-liquidai-vllm"
            ;;
        38|h200_meta_sglang|h200-meta-sglang)
            echo "h200-meta-sglang"
            ;;
        39|h200_meta_vllm|h200-meta-vllm)
            echo "h200-meta-vllm"
            ;;
        40|h200_microsoft_vllm|h200-microsoft-vllm)
            echo "h200-microsoft-vllm"
            ;;
        41|h200_minimax_m2_sglang_v0510_post1|h200-minimax-m2-sglang-v0510-post1)
            echo "h200-minimax-m2-sglang-v0510-post1"
            ;;
        42|h200_minimax_m2_vllm_0f3ce4c74|h200-minimax-m2-vllm-0f3ce4c74)
            echo "h200-minimax-m2-vllm-0f3ce4c74"
            ;;
        43|h200_minimax_m25_vllm_v0280|h200-minimax-m25-vllm-v0280)
            echo "h200-minimax-m25-vllm-v0280"
            ;;
        44|h200_mistralai_sglang|h200-mistralai-sglang)
            echo "h200-mistralai-sglang"
            ;;
        45|h200_mistralai_vllm|h200-mistralai-vllm)
            echo "h200-mistralai-vllm"
            ;;
        46|h200_nanbeige_sglang|h200-nanbeige-sglang)
            echo "h200-nanbeige-sglang"
            ;;
        47|h200_nanbeige_vllm|h200-nanbeige-vllm)
            echo "h200-nanbeige-vllm"
            ;;
        48|h200_nemotron_ultra_vllm_9c2d21046|h200-nemotron-ultra-vllm-9c2d21046)
            echo "h200-nemotron-ultra-vllm-9c2d21046"
            ;;
        49|h200_nex_n2_sglang_v0519|h200-nex-n2-sglang-v0519)
            echo "h200-nex-n2-sglang-v0519"
            ;;
        50|h200_nex_n2_vllm_v0290|h200-nex-n2-vllm-v0290)
            echo "h200-nex-n2-vllm-v0290"
            ;;
        51|h200_nvidia_deepseek_sglang|h200-nvidia-deepseek-sglang)
            echo "h200-nvidia-deepseek-sglang"
            ;;
        52|h200_nvidia_glm53_sglang_26fd7fd|h200-nvidia-glm53-sglang-26fd7fd)
            echo "h200-nvidia-glm53-sglang-26fd7fd"
            ;;
        53|h200_nvidia_glm53flash_dflash2_sglang_4a0c6f2fe|h200-nvidia-glm53flash-dflash2-sglang-4a0c6f2fe)
            echo "h200-nvidia-glm53flash-dflash2-sglang-4a0c6f2fe"
            ;;
        54|h200_nvidia_glm53flash_dflash2_vllm_3b5189828|h200-nvidia-glm53flash-dflash2-vllm-3b5189828)
            echo "h200-nvidia-glm53flash-dflash2-vllm-3b5189828"
            ;;
        55|h200_nvidia_muse_sglang_v0520|h200-nvidia-muse-sglang-v0520)
            echo "h200-nvidia-muse-sglang-v0520"
            ;;
        56|h200_nvidia_muse_vllm_v0290|h200-nvidia-muse-vllm-v0290)
            echo "h200-nvidia-muse-vllm-v0290"
            ;;
        57|h200_nvidia_nemotron|h200-nvidia-nemotron)
            echo "h200-nvidia-nemotron"
            ;;
        58|h200_nvidia_sglang|h200-nvidia-sglang)
            echo "h200-nvidia-sglang"
            ;;
        59|h200_nvidia_sglang_pr_33554|h200-nvidia-sglang-pr-33554)
            echo "h200-nvidia-sglang-pr-33554"
            ;;
        60|h200_nvidia_vllm|h200-nvidia-vllm)
            echo "h200-nvidia-vllm"
            ;;
        61|h200_nvidia_vllm_pr_55222|h200-nvidia-vllm-pr-55222)
            echo "h200-nvidia-vllm-pr-55222"
            ;;
        62|h200_openai_sglang_pr_38626|h200-openai-sglang-pr-38626)
            echo "h200-openai-sglang-pr-38626"
            ;;
        63|h200_openai_vllm_pr_53207|h200-openai-vllm-pr-53207)
            echo "h200-openai-vllm-pr-53207"
            ;;
        64|h200_paradigma_inc_vllm_v0260|h200-paradigma-inc-vllm-v0260)
            echo "h200-paradigma-inc-vllm-v0260"
            ;;
        65|h200_poolside_laguna_xs_vllm|h200-poolside-laguna-xs-vllm)
            echo "h200-poolside-laguna-xs-vllm"
            ;;
        66|h200_poolside_sglang|h200-poolside-sglang)
            echo "h200-poolside-sglang"
            ;;
        67|h200_poolside_vllm|h200-poolside-vllm)
            echo "h200-poolside-vllm"
            ;;
        68|h200_primeintellect_sglang|h200-primeintellect-sglang)
            echo "h200-primeintellect-sglang"
            ;;
        69|h200_primeintellect_vllm|h200-primeintellect-vllm)
            echo "h200-primeintellect-vllm"
            ;;
        70|h200_qwen_flash_next_sglang|h200-qwen-flash-next-sglang)
            echo "h200-qwen-flash-next-sglang"
            ;;
        71|h200_qwen_flash_next_vllm|h200-qwen-flash-next-vllm)
            echo "h200-qwen-flash-next-vllm"
            ;;
        72|h200_qwen_flash_next_vllm_pr_54129|h200-qwen-flash-next-vllm-pr-54129)
            echo "h200-qwen-flash-next-vllm-pr-54129"
            ;;
        73|h200_qwen_sglang|h200-qwen-sglang)
            echo "h200-qwen-sglang"
            ;;
        74|h200_qwen_sglang_pr_22121|h200-qwen-sglang-pr-22121)
            echo "h200-qwen-sglang-pr-22121"
            ;;
        75|h200_qwen_vllm|h200-qwen-vllm)
            echo "h200-qwen-vllm"
            ;;
        76|h200_radixark_qwen_sglang|h200-radixark-qwen-sglang)
            echo "h200-radixark-qwen-sglang"
            ;;
        77|h200_radixark_sglang|h200-radixark-sglang)
            echo "h200-radixark-sglang"
            ;;
        78|h200_redhat_sglang_pr_35809|h200-redhat-sglang-pr-35809)
            echo "h200-redhat-sglang-pr-35809"
            ;;
        79|h200_redhatai_sglang|h200-redhatai-sglang)
            echo "h200-redhatai-sglang"
            ;;
        80|h200_redhatai_vllm|h200-redhatai-vllm)
            echo "h200-redhatai-vllm"
            ;;
        81|h200_stepfun_sglang|h200-stepfun-sglang)
            echo "h200-stepfun-sglang"
            ;;
        82|h200_stepfun_vllm|h200-stepfun-vllm)
            echo "h200-stepfun-vllm"
            ;;
        83|h200_xiaomimimo_flash_vllm_1ea7c63|h200-xiaomimimo-flash-vllm-1ea7c63)
            echo "h200-xiaomimimo-flash-vllm-1ea7c63"
            ;;
        84|h200_xiaomimimo_sglang_v0520|h200-xiaomimimo-sglang-v0520)
            echo "h200-xiaomimimo-sglang-v0520"
            ;;
        85|h200_xiaomimimo_vllm_v0300|h200-xiaomimimo-vllm-v0300)
            echo "h200-xiaomimimo-vllm-v0300"
            ;;
        86|h200_z_lab_sglang|h200-z-lab-sglang)
            echo "h200-z-lab-sglang"
            ;;
        87|h200_z_lab_sglang_pr_35209|h200-z-lab-sglang-pr-35209)
            echo "h200-z-lab-sglang-pr-35209"
            ;;
        88|h200_z_lab_vllm|h200-z-lab-vllm)
            echo "h200-z-lab-vllm"
            ;;
        89|h200_zyphra_legacy_vllm|h200-zyphra-legacy-vllm)
            echo "h200-zyphra-legacy-vllm"
            ;;
        90|h200_zyphra_sglang|h200-zyphra-sglang)
            echo "h200-zyphra-sglang"
            ;;
        91|h200_zyphra_sglang_pr_32517|h200-zyphra-sglang-pr-32517)
            echo "h200-zyphra-sglang-pr-32517"
            ;;
        92|h200_zyphra_vllm|h200-zyphra-vllm)
            echo "h200-zyphra-vllm"
            ;;
        93|rtxpro6k_allenai_vllm_73a5831127a9|rtxpro6k-allenai-vllm-73a5831127a9)
            echo "rtxpro6k-allenai-vllm-73a5831127a9"
            ;;
        94|rtxpro6k_allenai_vllm_73a78e6f1f38|rtxpro6k-allenai-vllm-73a78e6f1f38)
            echo "rtxpro6k-allenai-vllm-73a78e6f1f38"
            ;;
        95|rtxpro6k_arcee_ai_vllm_73a5831127a9|rtxpro6k-arcee-ai-vllm-73a5831127a9)
            echo "rtxpro6k-arcee-ai-vllm-73a5831127a9"
            ;;
        96|rtxpro6k_arcee_ai_vllm_c8414a82712b|rtxpro6k-arcee-ai-vllm-c8414a82712b)
            echo "rtxpro6k-arcee-ai-vllm-c8414a82712b"
            ;;
        97|rtxpro6k_coherelabs_vllm_73a5831127a9|rtxpro6k-coherelabs-vllm-73a5831127a9)
            echo "rtxpro6k-coherelabs-vllm-73a5831127a9"
            ;;
        98|rtxpro6k_datalab_to_vllm_73a5831127a9|rtxpro6k-datalab-to-vllm-73a5831127a9)
            echo "rtxpro6k-datalab-to-vllm-73a5831127a9"
            ;;
        99|rtxpro6k_deepseek_v41_1017_vllm_pr_60762|rtxpro6k-deepseek-v41-1017-vllm-pr-60762)
            echo "rtxpro6k-deepseek-v41-1017-vllm-pr-60762"
            ;;
        100|rtxpro6k_deepseek_v41_vllm_pr_56509|rtxpro6k-deepseek-v41-vllm-pr-56509)
            echo "rtxpro6k-deepseek-v41-vllm-pr-56509"
            ;;
        101|rtxpro6k_glm53_vllm_2617fe938|rtxpro6k-glm53-vllm-2617fe938)
            echo "rtxpro6k-glm53-vllm-2617fe938"
            ;;
        102|rtxpro6k_google_sglang_1093c501dfef|rtxpro6k-google-sglang-1093c501dfef)
            echo "rtxpro6k-google-sglang-1093c501dfef"
            ;;
        103|rtxpro6k_google_vllm_73a5831127a9|rtxpro6k-google-vllm-73a5831127a9)
            echo "rtxpro6k-google-vllm-73a5831127a9"
            ;;
        104|rtxpro6k_ibm_granite_sglang_fa7784d140ee|rtxpro6k-ibm-granite-sglang-fa7784d140ee)
            echo "rtxpro6k-ibm-granite-sglang-fa7784d140ee"
            ;;
        105|rtxpro6k_ibm_granite_vllm_73a5831127a9|rtxpro6k-ibm-granite-vllm-73a5831127a9)
            echo "rtxpro6k-ibm-granite-vllm-73a5831127a9"
            ;;
        106|rtxpro6k_inclusionai_sglang_1093c501dfef|rtxpro6k-inclusionai-sglang-1093c501dfef)
            echo "rtxpro6k-inclusionai-sglang-1093c501dfef"
            ;;
        107|rtxpro6k_inclusionai_vllm_73a5831127a9|rtxpro6k-inclusionai-vllm-73a5831127a9)
            echo "rtxpro6k-inclusionai-vllm-73a5831127a9"
            ;;
        108|rtxpro6k_incoai_sglang_964c45cf3|rtxpro6k-incoai-sglang-964c45cf3)
            echo "rtxpro6k-incoai-sglang-964c45cf3"
            ;;
        109|rtxpro6k_incoai_sglang_fa7784d140ee|rtxpro6k-incoai-sglang-fa7784d140ee)
            echo "rtxpro6k-incoai-sglang-fa7784d140ee"
            ;;
        110|rtxpro6k_incoai_sglang_pr_028ac64f7|rtxpro6k-incoai-sglang-pr-028ac64f7)
            echo "rtxpro6k-incoai-sglang-pr-028ac64f7"
            ;;
        111|rtxpro6k_incoai_vllm_73a583112|rtxpro6k-incoai-vllm-73a583112)
            echo "rtxpro6k-incoai-vllm-73a583112"
            ;;
        112|rtxpro6k_incoai_vllm_pr_417b0b6aa|rtxpro6k-incoai-vllm-pr-417b0b6aa)
            echo "rtxpro6k-incoai-vllm-pr-417b0b6aa"
            ;;
        113|rtxpro6k_intel_sglang|rtxpro6k-intel-sglang)
            echo "rtxpro6k-intel-sglang"
            ;;
        114|rtxpro6k_intel_vllm_73a5831127a9|rtxpro6k-intel-vllm-73a5831127a9)
            echo "rtxpro6k-intel-vllm-73a5831127a9"
            ;;
        115|rtxpro6k_liquidai_sglang|rtxpro6k-liquidai-sglang)
            echo "rtxpro6k-liquidai-sglang"
            ;;
        116|rtxpro6k_liquidai_sglang_pr_31041|rtxpro6k-liquidai-sglang-pr-31041)
            echo "rtxpro6k-liquidai-sglang-pr-31041"
            ;;
        117|rtxpro6k_meta_models_sglang_fa7784d140ee|rtxpro6k-meta-models-sglang-fa7784d140ee)
            echo "rtxpro6k-meta-models-sglang-fa7784d140ee"
            ;;
        118|rtxpro6k_meta_models_vllm_73a5831127a9|rtxpro6k-meta-models-vllm-73a5831127a9)
            echo "rtxpro6k-meta-models-vllm-73a5831127a9"
            ;;
        119|rtxpro6k_microsoft_vllm_73a5831127a9|rtxpro6k-microsoft-vllm-73a5831127a9)
            echo "rtxpro6k-microsoft-vllm-73a5831127a9"
            ;;
        120|rtxpro6k_minimaxai_vllm_e0317b28ce3b|rtxpro6k-minimaxai-vllm-e0317b28ce3b)
            echo "rtxpro6k-minimaxai-vllm-e0317b28ce3b"
            ;;
        121|rtxpro6k_mistralai_sglang_fa7784d140ee|rtxpro6k-mistralai-sglang-fa7784d140ee)
            echo "rtxpro6k-mistralai-sglang-fa7784d140ee"
            ;;
        122|rtxpro6k_mistralai_vllm_73a5831127a9|rtxpro6k-mistralai-vllm-73a5831127a9)
            echo "rtxpro6k-mistralai-vllm-73a5831127a9"
            ;;
        123|rtxpro6k_mistralai_vllm_pr_678ef33584da|rtxpro6k-mistralai-vllm-pr-678ef33584da)
            echo "rtxpro6k-mistralai-vllm-pr-678ef33584da"
            ;;
        124|rtxpro6k_moonshotai_vllm_73a5831127a9|rtxpro6k-moonshotai-vllm-73a5831127a9)
            echo "rtxpro6k-moonshotai-vllm-73a5831127a9"
            ;;
        125|rtxpro6k_nanbeige_sglang|rtxpro6k-nanbeige-sglang)
            echo "rtxpro6k-nanbeige-sglang"
            ;;
        126|rtxpro6k_nanbeige_vllm_pr_62f6de733d7a|rtxpro6k-nanbeige-vllm-pr-62f6de733d7a)
            echo "rtxpro6k-nanbeige-vllm-pr-62f6de733d7a"
            ;;
        127|rtxpro6k_nex_agi_vllm_e0317b28ce3b|rtxpro6k-nex-agi-vllm-e0317b28ce3b)
            echo "rtxpro6k-nex-agi-vllm-e0317b28ce3b"
            ;;
        128|rtxpro6k_nvidia_glm53_sglang_26fd7fd|rtxpro6k-nvidia-glm53-sglang-26fd7fd)
            echo "rtxpro6k-nvidia-glm53-sglang-26fd7fd"
            ;;
        129|rtxpro6k_nvidia_glm53flash_dflash2_sglang_pr_38430|rtxpro6k-nvidia-glm53flash-dflash2-sglang-pr-38430)
            echo "rtxpro6k-nvidia-glm53flash-dflash2-sglang-pr-38430"
            ;;
        130|rtxpro6k_nvidia_glm53flash_dflash2_vllm_8c60cf07b|rtxpro6k-nvidia-glm53flash-dflash2-vllm-8c60cf07b)
            echo "rtxpro6k-nvidia-glm53flash-dflash2-vllm-8c60cf07b"
            ;;
        131|rtxpro6k_nvidia_glm53flash_sglang_pr_38430|rtxpro6k-nvidia-glm53flash-sglang-pr-38430)
            echo "rtxpro6k-nvidia-glm53flash-sglang-pr-38430"
            ;;
        132|rtxpro6k_nvidia_qwen38_vllm_9c2d21046|rtxpro6k-nvidia-qwen38-vllm-9c2d21046)
            echo "rtxpro6k-nvidia-qwen38-vllm-9c2d21046"
            ;;
        133|rtxpro6k_nvidia_sglang_964c45cf3|rtxpro6k-nvidia-sglang-964c45cf3)
            echo "rtxpro6k-nvidia-sglang-964c45cf3"
            ;;
        134|rtxpro6k_nvidia_sglang_pr_13e5dcf04f082f8040f46d49|rtxpro6k-nvidia-sglang-pr-13e5dcf04f082f8040f46d49)
            echo "rtxpro6k-nvidia-sglang-pr-13e5dcf04f082f8040f46d49"
            ;;
        135|rtxpro6k_nvidia_vllm_73a5831127a9|rtxpro6k-nvidia-vllm-73a5831127a9)
            echo "rtxpro6k-nvidia-vllm-73a5831127a9"
            ;;
        136|rtxpro6k_nvidia_vllm_pr_39e0ce172|rtxpro6k-nvidia-vllm-pr-39e0ce172)
            echo "rtxpro6k-nvidia-vllm-pr-39e0ce172"
            ;;
        137|rtxpro6k_nvidia_vllm_pr_9fedd24fa864|rtxpro6k-nvidia-vllm-pr-9fedd24fa864)
            echo "rtxpro6k-nvidia-vllm-pr-9fedd24fa864"
            ;;
        138|rtxpro6k_openai_vllm_73a5831127a9|rtxpro6k-openai-vllm-73a5831127a9)
            echo "rtxpro6k-openai-vllm-73a5831127a9"
            ;;
        139|rtxpro6k_paradigma_inc_vllm_73a5831127a9|rtxpro6k-paradigma-inc-vllm-73a5831127a9)
            echo "rtxpro6k-paradigma-inc-vllm-73a5831127a9"
            ;;
        140|rtxpro6k_poolside_vllm_73a5831127a9|rtxpro6k-poolside-vllm-73a5831127a9)
            echo "rtxpro6k-poolside-vllm-73a5831127a9"
            ;;
        141|rtxpro6k_primeintellect_sglang|rtxpro6k-primeintellect-sglang)
            echo "rtxpro6k-primeintellect-sglang"
            ;;
        142|rtxpro6k_primeintellect_vllm_73a5831127a9|rtxpro6k-primeintellect-vllm-73a5831127a9)
            echo "rtxpro6k-primeintellect-vllm-73a5831127a9"
            ;;
        143|rtxpro6k_qwen_flash_next_vllm|rtxpro6k-qwen-flash-next-vllm)
            echo "rtxpro6k-qwen-flash-next-vllm"
            ;;
        144|rtxpro6k_qwen_flash_next_vllm_pr_54129|rtxpro6k-qwen-flash-next-vllm-pr-54129)
            echo "rtxpro6k-qwen-flash-next-vllm-pr-54129"
            ;;
        145|rtxpro6k_qwen_sglang|rtxpro6k-qwen-sglang)
            echo "rtxpro6k-qwen-sglang"
            ;;
        146|rtxpro6k_qwen_vllm_73a5831127a9|rtxpro6k-qwen-vllm-73a5831127a9)
            echo "rtxpro6k-qwen-vllm-73a5831127a9"
            ;;
        147|rtxpro6k_radixark_sglang_fa7784d140ee|rtxpro6k-radixark-sglang-fa7784d140ee)
            echo "rtxpro6k-radixark-sglang-fa7784d140ee"
            ;;
        148|rtxpro6k_redhatai_vllm_73a5831127a9d2b8|rtxpro6k-redhatai-vllm-73a5831127a9d2b8)
            echo "rtxpro6k-redhatai-vllm-73a5831127a9d2b8"
            ;;
        149|rtxpro6k_stepfun_ai_sglang_f92199b569d2|rtxpro6k-stepfun-ai-sglang-f92199b569d2)
            echo "rtxpro6k-stepfun-ai-sglang-f92199b569d2"
            ;;
        150|rtxpro6k_xiaomimimo_flash_vllm_pr_58177|rtxpro6k-xiaomimimo-flash-vllm-pr-58177)
            echo "rtxpro6k-xiaomimimo-flash-vllm-pr-58177"
            ;;
        151|rtxpro6k_z_lab_vllm_73a5831127a9|rtxpro6k-z-lab-vllm-73a5831127a9)
            echo "rtxpro6k-z-lab-vllm-73a5831127a9"
            ;;
        152|rtxpro6k_z_lab_vllm_pr_e55909cad00f|rtxpro6k-z-lab-vllm-pr-e55909cad00f)
            echo "rtxpro6k-z-lab-vllm-pr-e55909cad00f"
            ;;
        153|rtxpro6k_zyphra_vllm_pr_2ce2f26c282d|rtxpro6k-zyphra-vllm-pr-2ce2f26c282d)
            echo "rtxpro6k-zyphra-vllm-pr-2ce2f26c282d"
            ;;
        154|rtxpro6k_zyphra_vllm_pr_b1b99a08b20f|rtxpro6k-zyphra-vllm-pr-b1b99a08b20f)
            echo "rtxpro6k-zyphra-vllm-pr-b1b99a08b20f"
            ;;
        155|custom|custom_uv|custom-uv|env_custom_uv)
            echo "custom_uv"
            ;;
        156|custom_pip|custom-pip|env_custom_pip)
            echo "custom_pip"
            ;;
        *)
            return 1
            ;;
    esac
}

resolve_env_name() {
    local env_type="$1"

    if [ -z "$env_type" ]; then
        echo "custom_uv"
        return 0
    fi

    echo "$env_type"
}

# This launcher mutates the caller's shell and therefore must be sourced.
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
    print_error "This script must be sourced, not executed!"
    print_error "Use: source $0 [--auto] [ENV_NAME|1-156]"
    exit 1
fi

# Parse arguments
AUTO_MODE=false
ENV_TYPE=""

for arg in "$@"; do
    case $arg in
        --auto)
            AUTO_MODE=true
            ;;
        *)
            if [[ ! "$arg" =~ ^-- ]]; then
                ENV_TYPE="$arg"
            fi
            ;;
    esac
done

# Prompt for environment type if not provided and not in auto mode
if [ -z "$ENV_TYPE" ] && [ "$AUTO_MODE" = false ]; then
    echo ""
    print_info "Select ML environment type:"
    echo "1) H200 AllenAI (vLLM)"
    echo "2) H200 Arcee NVFP4 (vLLM)"
    echo "3) H200 Arcee (vLLM)"
    echo "4) H200 Arcee (vLLM) PR 54479"
    echo "5) H200 Arcee FP8 Block (vLLM) PR 54479"
    echo "6) H200 Arcee Thinking FP8 Block (vLLM) PR 54479"
    echo "7) H200 Cohere (vLLM)"
    echo "8) H200 Cohere (vLLM) PR 54479"
    echo "9) H200 DataLab (vLLM)"
    echo "10) H200 DeepSeek (SGLang)"
    echo "11) H200 DeepSeek V4.1 Flash 1017 (vLLM) 5df63a720"
    echo "12) H200 DeepSeek V4.1 Flash (vLLM) e77daef89"
    echo "13) H200 DeepSeek V4 Flash Vision Exp (SGLang) PR 37253"
    echo "14) H200 DeepSeek V4 Flash Vision Exp (vLLM) PR 54566"
    echo "15) H200 DeepSeek (vLLM)"
    echo "16) H200 DiffusionGemma (SGLang)"
    echo "17) H200 Gemma (SGLang)"
    echo "18) H200 Gemma (vLLM)"
    echo "19) H200 Gemma 3n (vLLM 0.10)"
    echo "20) H200 GLM 5.3 (vLLM 0.29.0)"
    echo "21) H200 GLM 5.3 Flash DFlash2 (SGLang) PR 37818"
    echo "22) H200 GLM 5.3 Flash DFlash2 (vLLM) PR 55423"
    echo "23) H200 GLM 5.3 Flash (vLLM) PR 53906 merge"
    echo "24) H200 GPT-OSS (SGLang)"
    echo "25) H200 gpt-oss (vLLM)"
    echo "26) H200 IBM (SGLang)"
    echo "27) H200 IBM (vLLM)"
    echo "28) H200 InclusionAI Ling 3 (vLLM)"
    echo "29) H200 InclusionAI (SGLang)"
    echo "30) H200 InclusionAI (vLLM)"
    echo "31) H200 IncoAI (SGLang)"
    echo "32) H200 IncoAI (vLLM)"
    echo "33) H200 Intel (SGLang)"
    echo "34) H200 Intel (vLLM)"
    echo "35) H200 LiquidAI (SGLang)"
    echo "36) H200 LiquidAI (SGLang) PR 31041"
    echo "37) H200 LiquidAI (vLLM)"
    echo "38) H200 Meta (SGLang)"
    echo "39) H200 Meta (vLLM)"
    echo "40) H200 Microsoft (vLLM)"
    echo "41) H200 MiniMax M2 family (SGLang) 0.5.10.post1"
    echo "42) H200 MiniMax M2 family (vLLM) 0f3ce4c74"
    echo "43) H200 MiniMax M2.5 (vLLM) 0.28.0"
    echo "44) H200 MistralAI (SGLang)"
    echo "45) H200 MistralAI (vLLM)"
    echo "46) H200 Nanbeige (SGLang)"
    echo "47) H200 Nanbeige (vLLM)"
    echo "48) H200 NVIDIA Nemotron Ultra (vLLM) PR54788 merge"
    echo "49) H200 Nex N2 (SGLang) 0.5.19"
    echo "50) H200 Nex N2 (vLLM) 0.29.0"
    echo "51) H200 NVIDIA DeepSeek (SGLang)"
    echo "52) H200 NVIDIA GLM-5.3 NVFP4 (SGLang 26fd7fd)"
    echo "53) H200 NVIDIA GLM-5.3 Flash NVFP4 DFlash2 (SGLang 4a0c6f2fe)"
    echo "54) H200 NVIDIA GLM-5.3 Flash NVFP4 DFlash2 (vLLM 3b5189828)"
    echo "55) H200 NVIDIA Muse Glimmer (SGLang) 0.5.20"
    echo "56) H200 NVIDIA Muse Glimmer (vLLM) 0.29.0"
    echo "57) H200 NVIDIA Nemotron (vLLM)"
    echo "58) H200 NVIDIA (SGLang)"
    echo "59) H200 NVIDIA (SGLang) PR 33554"
    echo "60) H200 NVIDIA (vLLM)"
    echo "61) H200 NVIDIA GLM 5.3 Flash (vLLM) PR 55222"
    echo "62) H200 OpenAI Whisper (SGLang) PR 38626"
    echo "63) H200 OpenAI Whisper (vLLM) PR 53207"
    echo "64) H200 Paradigma Limite (vLLM 0.26.0, official plugin)"
    echo "65) H200 Poolside Laguna XS (vLLM)"
    echo "66) H200 Poolside (SGLang)"
    echo "67) H200 Poolside (vLLM)"
    echo "68) H200 PrimeIntellect (SGLang)"
    echo "69) H200 PrimeIntellect (vLLM)"
    echo "70) H200 Qwen Flash Next (SGLang)"
    echo "71) H200 Qwen Flash Next (vLLM)"
    echo "72) H200 Qwen Flash Next disk PLE (vLLM PR 54129)"
    echo "73) H200 Qwen (SGLang)"
    echo "74) H200 Qwen (SGLang) PR 22121"
    echo "75) H200 Qwen (vLLM)"
    echo "76) H200 RadixArk Qwen Flash Next (SGLang)"
    echo "77) H200 RadixArk (SGLang)"
    echo "78) H200 RedHat (SGLang) PR 35809"
    echo "79) H200 RedHatAI (SGLang)"
    echo "80) H200 RedHatAI (vLLM)"
    echo "81) H200 StepFun (SGLang)"
    echo "82) H200 StepFun (vLLM)"
    echo "83) H200 XiaomiMiMo Flash (vLLM 1ea7c63, audio)"
    echo "84) H200 XiaomiMiMo Distill (SGLang 0.5.20)"
    echo "85) H200 XiaomiMiMo Distill (vLLM 0.30.0)"
    echo "86) H200 z-lab (SGLang)"
    echo "87) H200 z-lab (SGLang) PR 35209"
    echo "88) H200 z-lab (vLLM)"
    echo "89) H200 Zyphra Legacy (vLLM)"
    echo "90) H200 Zyphra (SGLang)"
    echo "91) H200 Zyphra (SGLang) PR 32517"
    echo "92) H200 Zyphra (vLLM)"
    echo "93) RTX PRO 6000 AllenAI (vLLM) 73a5831127a9"
    echo "94) RTX PRO 6000 AllenAI (vLLM) 73a78e6f1f38"
    echo "95) RTX PRO 6000 Arcee AI (vLLM) 73a5831127a9"
    echo "96) RTX PRO 6000 Arcee AI (vLLM) c8414a82712b NVFP4"
    echo "97) RTX PRO 6000 CohereLabs (vLLM) 73a5831127a9"
    echo "98) RTX PRO 6000 Datalab (vLLM) 73a5831127a9"
    echo "99) RTX PRO 6000 DeepSeek V4.1 Flash 1017 (vLLM) PR 60762 SM120"
    echo "100) RTX PRO 6000 DeepSeek V4.1 Flash (vLLM) PR 56509 SM120"
    echo "101) RTX PRO 6000 GLM 5.3 Flash (vLLM) 2617fe938 SM120"
    echo "102) RTX PRO 6000 Google Gemma 4 (SGLang 1093c501dfef)"
    echo "103) RTX PRO 6000 Google Gemma (vLLM 73a5831127a9, audio)"
    echo "104) RTX PRO 6000 IBM Granite (SGLang fa7784d140ee)"
    echo "105) RTX PRO 6000 IBM Granite (vLLM 73a5831127a9)"
    echo "106) RTX PRO 6000 Inclusion Ling (SGLang 1093c501dfef)"
    echo "107) RTX PRO 6000 Inclusion Ling (vLLM 73a5831127a9)"
    echo "108) RTX PRO 6000 incoai GLM-5.3 DFlash2 (SGLang 964c45cf3)"
    echo "109) RTX PRO 6000 incoai Qwen3.8 DFlash2 (SGLang fa7784d140ee)"
    echo "110) RTX PRO 6000 incoai GLM-5.3 Flash (SGLang PR 38430, 028ac64f7)"
    echo "111) RTX PRO 6000 incoai GLM-5.3 Flash (vLLM 73a583112, b12x)"
    echo "112) RTX PRO 6000 incoai GLM-5.3 DFlash2 (vLLM PR 58773, 417b0b6aa)"
    echo "113) RTX PRO 6000 Intel (SGLang)"
    echo "114) RTX PRO 6000 Intel AutoRound (vLLM 73a5831127a9)"
    echo "115) RTX PRO 6000 LiquidAI (SGLang)"
    echo "116) RTX PRO 6000 LiquidAI (SGLang) PR 31041"
    echo "117) RTX PRO 6000 Meta Muse (SGLang fa7784d140ee)"
    echo "118) RTX PRO 6000 Meta Muse (vLLM 73a5831127a9)"
    echo "119) RTX PRO 6000 Microsoft Fara (vLLM 73a5831127a9)"
    echo "120) RTX PRO 6000 MiniMaxAI (vLLM e0317b28ce3b)"
    echo "121) RTX PRO 6000 MistralAI (SGLang fa7784d140ee)"
    echo "122) RTX PRO 6000 MistralAI (vLLM 73a5831127a9)"
    echo "123) RTX PRO 6000 MistralAI (vLLM PR 58172, 678ef33584da)"
    echo "124) RTX PRO 6000 MoonshotAI (vLLM 73a5831127a9)"
    echo "125) RTX PRO 6000 Nanbeige (SGLang)"
    echo "126) RTX PRO 6000 Nanbeige (vLLM PR 49433, 62f6de733d7a)"
    echo "127) RTX PRO 6000 Nex AGI (vLLM e0317b28ce3b)"
    echo "128) RTX PRO 6000 NVIDIA GLM-5.3 NVFP4 (SGLang 26fd7fd)"
    echo "129) RTX PRO 6000 NVIDIA GLM-5.3 Flash NVFP4 DFlash2 (SGLang PR 38430, 4df49edcf)"
    echo "130) RTX PRO 6000 NVIDIA GLM-5.3 Flash NVFP4 DFlash2 (vLLM 8c60cf07b, cuBLAS 13.6.1)"
    echo "131) RTX PRO 6000 NVIDIA GLM-5.3 Flash NVFP4 (SGLang) PR 38430"
    echo "132) RTX PRO 6000 NVIDIA Qwen3.8 Flash Next NVFP4 (vLLM) 9c2d21046"
    echo "133) RTX PRO 6000 NVIDIA Kimi-K2.6 NVFP4 (SGLang 964c45cf3)"
    echo "134) RTX PRO 6000 NVIDIA Qwen3.6 NVFP4 (SGLang PR 41352, 13e5dcf04f082f8040f46d49)"
    echo "135) RTX PRO 6000 NVIDIA (vLLM 73a5831127a9)"
    echo "136) RTX PRO 6000 NVIDIA Kimi (vLLM PR 54013, 39e0ce172)"
    echo "137) RTX PRO 6000 NVIDIA Kimi (vLLM PR 54013, 9fedd24fa864)"
    echo "138) RTX PRO 6000 OpenAI (vLLM 73a5831127a9)"
    echo "139) RTX PRO 6000 Paradigma Limite (vLLM 73a5831127a9, official plugin)"
    echo "140) RTX PRO 6000 Poolside (vLLM 73a5831127a9)"
    echo "141) RTX PRO 6000 PrimeIntellect (SGLang)"
    echo "142) RTX PRO 6000 PrimeIntellect (vLLM 73a5831127a9)"
    echo "143) RTX PRO 6000 Qwen Flash Next (vLLM)"
    echo "144) RTX PRO 6000 Qwen Flash Next disk PLE (vLLM PR 54129)"
    echo "145) RTX PRO 6000 Qwen (SGLang)"
    echo "146) RTX PRO 6000 Qwen (vLLM 73a5831127a9)"
    echo "147) RTX PRO 6000 RadixArk (SGLang fa7784d140ee)"
    echo "148) RTX PRO 6000 RedHatAI (vLLM 73a5831127a9d2b8, timm)"
    echo "149) RTX PRO 6000 StepFun AI (SGLang f92199b569d2)"
    echo "150) RTX PRO 6000 XiaomiMiMo Flash (vLLM PR 58177, audio)"
    echo "151) RTX PRO 6000 z-lab (vLLM 73a5831127a9)"
    echo "152) RTX PRO 6000 z-lab DFlash2 (vLLM PR 58207, e55909cad00f)"
    echo "153) RTX PRO 6000 Zyphra ZAYA legacy (vLLM PR 41999, 2ce2f26c282d)"
    echo "154) RTX PRO 6000 Zyphra ZAYA current (vLLM PR 41999, b1b99a08b20f)"
    echo "155) Custom (uv)"
    echo "156) Custom (pip)"
    echo ""
    while true; do
        read -r -p "Enter your choice (1-156): " choice
        if ENV_TYPE=$(resolve_env_type "$choice"); then
            break
        else
            print_error "Invalid choice. Please enter a number between 1 and 156."
        fi
    done
elif [ -z "$ENV_TYPE" ]; then
    # Default to Custom (uv) in auto mode
    ENV_TYPE="custom_uv"
fi

# Normalize and validate the selected managed environment.
if [ -n "$ENV_TYPE" ]; then
    if ! ENV_TYPE_MAPPED=$(resolve_env_type "$ENV_TYPE"); then
        print_error "Invalid environment selection: $ENV_TYPE. Choose a listed environment name or a number from 1 to 156."
        return 1
    fi
    ENV_TYPE="$ENV_TYPE_MAPPED"
fi

# Set environment name based on type
ENV_NAME=$(resolve_env_name "$ENV_TYPE")

ENV_PATH="$HOME/env_${ENV_NAME}"

# Check if environment exists
if [ ! -d "$ENV_PATH" ]; then
    print_error "Environment '$ENV_NAME' not found at $ENV_PATH"
    print_info "Run ./installers/05_setup_env.sh first to create it"
    return 1
fi

# Check if already in a virtual environment
if [ -n "$VIRTUAL_ENV" ]; then
    print_warning "Already in virtual environment: $VIRTUAL_ENV"
    if [ "$AUTO_MODE" = false ]; then
        read -r -p "Deactivate and switch to $ENV_NAME? (y/n): " SWITCH
    else
        SWITCH="y"
        print_info "Auto mode: switching environment"
    fi
    if [[ "$SWITCH" =~ ^[Yy]$ ]]; then
        deactivate
    else
        return 0
    fi
fi

# Check if activate_ml script exists, use it if available
if [ -f "$ENV_PATH/activate_ml" ]; then
    print_info "Using activate_ml script..."
    if ! source "$ENV_PATH/activate_ml"; then
        print_error "Failed to activate environment '$ENV_NAME' with $ENV_PATH/activate_ml"
        return 1
    fi
elif [ -f "$ENV_PATH/bin/activate" ]; then
    # Fallback to manual activation
    print_info "Activating $ENV_NAME environment..."
    if ! source "$ENV_PATH/bin/activate"; then
        print_error "Failed to activate environment '$ENV_NAME' with $ENV_PATH/bin/activate"
        return 1
    fi
    
    # Determine HF_PATH - check if already set, otherwise use default
    if [ -n "$HF_HOME" ]; then
        HF_PATH="$HF_HOME"
    else
        HF_PATH="$WORKSPACE_DIR/models/huggingface"
    fi
    
    # Set ML environment variables
    export HF_HOME="$HF_PATH"
    export HF_HUB_CACHE="$HF_PATH/hub"

else
    print_error "No activation script found for environment '$ENV_NAME'"
    print_info "Expected $ENV_PATH/activate_ml or $ENV_PATH/bin/activate"
    return 1
fi

if [ "${ML_ENV_CUDA_APPLIED:-}" != "1" ]; then
    if ! apply_env_cuda_selection; then
        return 1
    fi
fi

export HF_HUB_DISABLE_SHARED_BLOBS="${HF_HUB_DISABLE_SHARED_BLOBS:-1}"
export DG_JIT_CACHE_DIR="${VIRTUAL_ENV:-$ENV_PATH}/.cache/deep_gemm"
export FLASHINFER_WORKSPACE_BASE="${VIRTUAL_ENV:-$ENV_PATH}"
export SGLANG_CACHE_DIR="${VIRTUAL_ENV:-$ENV_PATH}/.cache/sglang"
export SGLANG_DG_CACHE_DIR="${VIRTUAL_ENV:-$ENV_PATH}/.cache/deep_gemm"
export TORCH_EXTENSIONS_DIR="${VIRTUAL_ENV:-$ENV_PATH}/.cache/torch_extensions"
export TORCH_HOME="${VIRTUAL_ENV:-$ENV_PATH}/.cache/torch"
export TORCHINDUCTOR_CACHE_DIR="${VIRTUAL_ENV:-$ENV_PATH}/.cache/torchinductor"
export TRITON_CACHE_DIR="${VIRTUAL_ENV:-$ENV_PATH}/.cache/triton"
export TRITON_HOME="${VIRTUAL_ENV:-$ENV_PATH}"
export TVM_FFI_CACHE_DIR="${VIRTUAL_ENV:-$ENV_PATH}/.cache/tvm-ffi"
export VLLM_CACHE_ROOT="${VIRTUAL_ENV:-$ENV_PATH}/.cache/vllm"
export VLLM_FLASHINFER_AUTOTUNE_CACHE_DIR="${VIRTUAL_ENV:-$ENV_PATH}/.cache/vllm/flashinfer_autotune_cache"
export XDG_CACHE_HOME="${VIRTUAL_ENV:-$ENV_PATH}/.cache"

# Display activation info if not using activate_ml script
if [ ! -f "$ENV_PATH/activate_ml" ]; then
    echo ""
    print_info "✓ ML environment activated!"
    echo "  - Virtual env: $ENV_PATH"
    echo "  - Python: $(which python) ($(python --version 2>&1))"
    echo "  - HF_HOME: $HF_HOME"
    echo "  - HF_HUB_CACHE: $HF_HUB_CACHE"
    echo "  - CPU threads: $OMP_NUM_THREADS"
    echo "  - TORCH_CUDA_ARCH_LIST: ${TORCH_CUDA_ARCH_LIST:-unset}"
    if [ -n "${ML_ENV_CUDA_HOME:-}" ]; then
        echo "  - CUDA toolkit: $ML_ENV_CUDA_HOME (${ML_ENV_CUDA_VERSION:-unknown}, ${ML_ENV_CUDA_SOURCE:-configured})"
    elif command -v nvcc &> /dev/null; then
        echo "  - CUDA: $(nvcc --version | grep release | awk '{print $6}')"
    fi
fi

echo ""
print_info "To deactivate: deactivate"
print_info "To install or configure environment packages: ./installers/06_install_packages.sh"