Spaces:

cpu4dream
/

llava-small-open-elm-aimv2

Runtime error

App Files Files Community

llava-small-open-elm-aimv2 / TinyLLaVA_Factory /scripts /train /train_mof.sh

Camil Ziane

init space

74b17e0 12 months ago

history blame contribute delete

870 Bytes

	DATA_PATH=/home/ai/data/llava/dataset/text_files/blip_laion_cc_sbu_558k.json
	FINETUNE_DATA_PATH=/home/ai/data/llava/dataset/text_files/llava_v1_5_mix665k.json
	IMAGE_PATH=/home/ai/data/llava/dataset/llava/llava_pretrain/images
	FINETUNE_IMAGE_PATH=/home/ai/data/llava/dataset

	LLM_VERSION=TinyLlama/TinyLlama-1.1B-Chat-v1.0
	VT_VERSION=mof:openai/clip-vit-large-patch14
	VT_VERSION2=mof:facebook/dinov2-large
	CN_VERSION=mof_mlp
	CONV_VERSION=llama
	VERSION=llama-mof-base
	TRAIN_RECIPE=common
	MODEL_MAX_LENGTH=2048


	bash scripts/train/pretrain.sh "$DATA_PATH" "$IMAGE_PATH" "$LLM_VERSION" "$VT_VERSION" "$VT_VERSION2" "$CN_VERSION" "$VERSION" "$TRAIN_RECIPE" "$MODEL_MAX_LENGTH"
	bash scripts/train/finetune.sh "$FINETUNE_DATA_PATH" "$FINETUNE_IMAGE_PATH" "$LLM_VERSION" "$VT_VERSION" "$VT_VERSION2" "$CN_VERSION" "$CONV_VERSION" "$VERSION" "$TRAIN_RECIPE" "$MODEL_MAX_LENGTH"