sjzhao commited on
Commit
e9e34da
1 Parent(s): 7dd51e9

update demo

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -0
  2. start.sh +1 -1
Dockerfile CHANGED
@@ -32,5 +32,7 @@ RUN git lfs install
32
 
33
  RUN pip install -r requirements.txt
34
 
 
 
35
  CMD ["bash", "start.sh"]
36
  # CMD ["python", 'start.py']
 
32
 
33
  RUN pip install -r requirements.txt
34
 
35
+ RUN ls -lh
36
+
37
  CMD ["bash", "start.sh"]
38
  # CMD ["python", 'start.py']
start.sh CHANGED
@@ -1,6 +1,6 @@
1
  python3 gradio_demo/seed_llama_flask.py \
2
  --image_transform configs/transform/clip_transform.yaml \
3
- --tokenizer configs/tokenizer/seed_llama_tokenizer.yaml \
4
  --model configs/llm/seed_llama_14b_8bit.yaml \
5
  --port 7890 \
6
  --llm_device cuda:0 \
 
1
  python3 gradio_demo/seed_llama_flask.py \
2
  --image_transform configs/transform/clip_transform.yaml \
3
+ --tokenizer configs/tokenizer/seed_llama_tokenizer_hf.yaml \
4
  --model configs/llm/seed_llama_14b_8bit.yaml \
5
  --port 7890 \
6
  --llm_device cuda:0 \