diff --git a/scripts/benchmark/setup.sh b/scripts/benchmark/setup.sh index 5bfd412..ba10d0e 100644 --- a/scripts/benchmark/setup.sh +++ b/scripts/benchmark/setup.sh @@ -60,7 +60,7 @@ fi log_info "Verifying GPU access in toolboxes..." for tb in "${REQUIRED_TOOLBOXES[@]}"; do if echo "$existing" | grep -qF "$tb"; then - if toolbox run -c "$tb" -- llama-cli --list-devices 2>&1 | grep -qi "gpu\|vulkan\|rocm"; then + if toolbox run -c "$tb" -- llama-cli --list-devices 2>/dev/null | grep -qi "vulkan\|rocm\|radeon\|available devices"; then log_success "GPU accessible in $tb" else log_warn "GPU may not be accessible in $tb — check device mappings" @@ -90,9 +90,8 @@ printf " ${BOLD}%-28s %-10s %-8s %s${RESET}\n" "Name" "Category" "Size" "Descri echo " $(printf '%.0s─' {1..70})" while IFS='|' read -r name repo file size_gb category desc; do [[ "$name" =~ ^#.*$ || -z "$name" ]] && continue - local_file="$MODEL_DIR/$file" status=" " - if [[ -f "$local_file" ]]; then + if find -L "$MODEL_DIR" -name "$file" -print -quit 2>/dev/null | grep -q .; then status="*" fi printf " %s %-27s %-10s %4s GB %s\n" "$status" "$name" "$category" "$size_gb" "$desc"