mattb512 commited on
Commit
4d27f4c
1 Parent(s): 98c0fb0
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. startup.sh +9 -9
Dockerfile CHANGED
@@ -6,7 +6,7 @@ RUN apt-get -y update \
6
  && rm -rf /var/lib/apt/lists/*
7
 
8
  RUN python3 -m pip install --upgrade pip \
9
- && pip install --no-cache-dir torch torchvision torchaudio pyyaml-include==1.4.1
10
 
11
  RUN useradd -m -u 1000 user
12
 
 
6
  && rm -rf /var/lib/apt/lists/*
7
 
8
  RUN python3 -m pip install --upgrade pip \
9
+ && pip install --no-cache-dir torch torchvision torchaudio pyyaml-include==1.4.1
10
 
11
  RUN useradd -m -u 1000 user
12
 
startup.sh CHANGED
@@ -15,13 +15,7 @@ python3 -c "import torch; \
15
 
16
  echo "Starting serve.controller"
17
  python3 -m serve.controller --host 0.0.0.0 --port 10000 &
18
- P1=$!
19
-
20
- sleep 30
21
-
22
- echo "Starting serve.gradio_web_server"
23
- python3 -m serve.gradio_web_server --controller http://127.0.0.1:10000 --model-list-mode reload & # --share
24
- P2=$!
25
 
26
  # sleep 30
27
 
@@ -33,7 +27,7 @@ sleep 30
33
 
34
  echo "Starting prism-dinosiglip+7b"
35
  python3 -m interactive_demo --port 40000 --model_id prism-dinosiglip+7b &
36
- P5=$!
37
 
38
  # python3 -m interactive_demo --port 40002 --model_id prism-dinosiglip-controlled+7b &
39
  # P6=$!
@@ -43,7 +37,13 @@ P5=$!
43
 
44
  sleep 30
45
 
 
 
 
 
 
 
46
  ls -als $HF_HOME
47
  tree --du -h $HF_HOME
48
 
49
- wait $P1 $P2 $P5 # $P5 $P6 $P7
 
15
 
16
  echo "Starting serve.controller"
17
  python3 -m serve.controller --host 0.0.0.0 --port 10000 &
18
+ P_CON=$!
 
 
 
 
 
 
19
 
20
  # sleep 30
21
 
 
27
 
28
  echo "Starting prism-dinosiglip+7b"
29
  python3 -m interactive_demo --port 40000 --model_id prism-dinosiglip+7b &
30
+ P_P7B=$!
31
 
32
  # python3 -m interactive_demo --port 40002 --model_id prism-dinosiglip-controlled+7b &
33
  # P6=$!
 
37
 
38
  sleep 30
39
 
40
+ echo "Starting serve.gradio_web_server"
41
+ python3 -m serve.gradio_web_server --controller http://127.0.0.1:10000 --model-list-mode reload & # --share
42
+ P_WEB=$!
43
+
44
+ sleep 30
45
+
46
  ls -als $HF_HOME
47
  tree --du -h $HF_HOME
48
 
49
+ wait $P_CON $P_WEB $P_P7B # $P5 $P6 $P7