| 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 | |