vlm-demo / startup.sh
mattb512's picture
gpu test
52494dd
raw
history blame
No virus
506 Bytes
#!/bin/bash
echo "PWD: $(pwd)"
echo $HF_TOKEN > .hf_token
echo "LS: $(ls -als)"
nvidia-smi
python -m serve.controller --host 0.0.0.0 --port 10000 &
P1=$!
sleep 30
python -m serve.gradio_web_server --controller http://127.0.0.1:10000 --model-list-mode reload --share &
P2=$!
python -m interactive_demo --port 40000 --model_id prism-dinosiglip+7b
# python -m interactive_demo --port 40001 --model_family llava-v15 --model_id llava-v1.5-7b --model_dir liuhaotian/llava-v1.5-7b
nvidia-smi
wait $P1 $P2