Spaces:
Running
on
A10G
Running
on
A10G
update demo
Browse files- Dockerfile +2 -0
- 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/
|
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 \
|