#!/bin/bash
set -e

MODELS_DIR="/nfsclient/os5800/vlm_project/data/pllum-vlm/models"


# LLaVA-PLLuM-12b-nc-instruct-250715
VIT_MODEL_NAME_OR_PATH="google/siglip2-so400m-patch14-384"
TEXT_MODEL_NAME_OR_PATH="CYFRAGOVPL/pllum-12b-nc-instruct-250715"
MODEL_NAME="pllum-12b-nc-instruct-250715-siglip2-so400m-patch14-384-llava-next"

# LLaVA-PLLuM-12b-nc-instruct
# VIT_MODEL_NAME_OR_PATH="google/siglip2-so400m-patch14-384"
# TEXT_MODEL_NAME_OR_PATH="CYFRAGOVPL/PLLuM-12B-nc-instruct"
# MODEL_NAME="pllum-12b-nc-instruct-siglip2-so400m-patch14-384-llava-next"

# LLaVA-Bielik-v2.6-11b-instruct
# VIT_MODEL_NAME_OR_PATH="google/siglip2-so400m-patch14-384"
# TEXT_MODEL_NAME_OR_PATH="speakleash/Bielik-11B-v2.6-Instruct"
# MODEL_NAME="bielik-11b-v2.6-instruct-siglip2-so400m-patch14-384-llava-next"

python -m src.modeling.llava.create_model \
    --vision_tower_model_name_or_path ${VIT_MODEL_NAME_OR_PATH} \
    --text_model_name_or_path ${TEXT_MODEL_NAME_OR_PATH} \
    --save_model_path "${MODELS_DIR}/base/${MODEL_NAME}" \
    --check_model True
