mattb512 commited on
Commit
3acf5d1
·
1 Parent(s): c7977b2
Files changed (1) hide show
  1. startup.sh +3 -3
startup.sh CHANGED
@@ -36,7 +36,7 @@ echo "$COMPONENT is running."
36
  #works!
37
  COMPONENT="interactive_demo prism-dinosiglip_7b"
38
  echo "Starting $COMPONENT"
39
- nohup python3 -m interactive_demo --port 40001 --model_id prism-dinosiglip+7b | tee prism-dinosiglip_7b.log &
40
  while [ `grep -c "loaded prismatic prism-dinosiglip+7b" prism-dinosiglip_7b.log` -eq '0' ];do
41
  sleep 10s;
42
  echo "waiting for $COMPONENT to be running"
@@ -46,7 +46,7 @@ echo "$COMPONENT is running."
46
  #works!
47
  COMPONENT="interactive_demo prism-dinosiglip-controlled+7b"
48
  echo "Starting $COMPONENT"
49
- nohup python3 -m interactive_demo --port 40002 --model_id prism-dinosiglip-controlled+7b | tee prism-dinosiglip-controlled_7b.log &
50
  while [ `grep -c "loaded prismatic prism-dinosiglip-controlled+7b" prism-dinosiglip-controlled_7b.log` -eq '0' ];do
51
  sleep 10s;
52
  echo "waiting for $COMPONENT to be running"
@@ -56,7 +56,7 @@ echo "$COMPONENT is running"
56
  #works!
57
  COMPONENT="interactive_demo llava-v1.5-7b"
58
  echo "Starting $COMPONENT"
59
- nohup python3 -m interactive_demo --port 40003 --model_family llava-v15 --model_id llava-v1.5-7b --model_dir liuhaotian/llava-v1.5-7b | tee llava-v1.5-7b.log &
60
  while [ `grep -c "loaded llava llava-v1.5-7b" llava-v1.5-7b.log` -eq '0' ];do
61
  sleep 10s;
62
  echo "waiting for $COMPONENT to be running"
 
36
  #works!
37
  COMPONENT="interactive_demo prism-dinosiglip_7b"
38
  echo "Starting $COMPONENT"
39
+ nohup CUDA_VISIBLE_DEVICES=0 python3 -m interactive_demo --port 40001 --model_id prism-dinosiglip+7b | tee prism-dinosiglip_7b.log &
40
  while [ `grep -c "loaded prismatic prism-dinosiglip+7b" prism-dinosiglip_7b.log` -eq '0' ];do
41
  sleep 10s;
42
  echo "waiting for $COMPONENT to be running"
 
46
  #works!
47
  COMPONENT="interactive_demo prism-dinosiglip-controlled+7b"
48
  echo "Starting $COMPONENT"
49
+ nohup CUDA_VISIBLE_DEVICES=1 python3 -m interactive_demo --port 40002 --model_id prism-dinosiglip-controlled+7b | tee prism-dinosiglip-controlled_7b.log &
50
  while [ `grep -c "loaded prismatic prism-dinosiglip-controlled+7b" prism-dinosiglip-controlled_7b.log` -eq '0' ];do
51
  sleep 10s;
52
  echo "waiting for $COMPONENT to be running"
 
56
  #works!
57
  COMPONENT="interactive_demo llava-v1.5-7b"
58
  echo "Starting $COMPONENT"
59
+ nohup CUDA_VISIBLE_DEVICES=2 python3 -m interactive_demo --port 40003 --model_family llava-v15 --model_id llava-v1.5-7b --model_dir liuhaotian/llava-v1.5-7b | tee llava-v1.5-7b.log &
60
  while [ `grep -c "loaded llava llava-v1.5-7b" llava-v1.5-7b.log` -eq '0' ];do
61
  sleep 10s;
62
  echo "waiting for $COMPONENT to be running"