#!/usr/bin/env bash
set -euo pipefail

BASE_MODEL="${BASE_MODEL:-checkpoints/llava-v1.6-vicuna-7b}"
PIXDLM_WEIGHT="${PIXDLM_WEIGHT:-pretrained/pixdlm-7b/pytorch_model.bin}"
SAVE_PATH="${SAVE_PATH:-pretrained/pixdlm-7b-merged}"

CUDA_VISIBLE_DEVICES="${CUDA_VISIBLE_DEVICES:-}" python merge_lora_weights_and_save_hf_model.py \
  --version="$BASE_MODEL" \
  --weight="$PIXDLM_WEIGHT" \
  --save_path="$SAVE_PATH" \
  --resize_vision_tower \
  --resize_vision_tower_size=448 \
  --vision_tower_for_mask \
  --Three_Level_Multi_Scale_Decoder