24 lines
777 B
Bash
24 lines
777 B
Bash
INPUT_CONFIG_FILE=input/config.json
|
|
|
|
CONFIG_FILE=configs/base.yaml
|
|
PATH_TO_VISION_ENCODER_MODEL=/weights/siglip-so400m-patch14-384
|
|
PATH_TO_TEXT_ENCODER_MODEL=/weights/t5-v1_1-xxl
|
|
PATH_TO_RDT_MODEL_WIGHTS=$(python3 scripts/read_config.py "$INPUT_CONFIG_FILE" evaluation.checkpoint_path)
|
|
DEVICE=cuda
|
|
CONTROL_FREQUENCY=25
|
|
PRE_LANG=False
|
|
LANG_MODEL=None
|
|
HOST=0.0.0.0
|
|
PORT=8000
|
|
|
|
python3 server.py \
|
|
--config_file $CONFIG_FILE \
|
|
--path_to_vision_encoder_model $PATH_TO_VISION_ENCODER_MODEL \
|
|
--path_to_text_encoder_model $PATH_TO_TEXT_ENCODER_MODEL \
|
|
--path_to_rdt_model_wights $PATH_TO_RDT_MODEL_WIGHTS \
|
|
--lang_model $LANG_MODEL \
|
|
--device $DEVICE \
|
|
--control_frequency $CONTROL_FREQUENCY \
|
|
--pre_lang $PRE_LANG \
|
|
--host $HOST \
|
|
--port $PORT |