first init

Dockerfile

@@ -0,0 +1,39 @@
+# Use the official Python 3.11 image
+FROM python:3.11
+
+# Set the working directory to /code
+WORKDIR /code
+
+RUN apt-get update && apt-get install -y screen git
+RUN git lfs install
+
+# Copy the current directory contents into the container at /code
+COPY ./requirements.txt /code/requirements.txt
+
+# Install requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
+# Switch to the "user" user
+USER user
+# Set home to the user's home directory
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
+
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
+
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+
+RUN git clone https://github.com/AILab-CVC/SEED.git
+RUN RUN mv SEED/* . && rm -rf SEED
+
+RUN git clone https://huggingface.co/AILab-CVC/SEED
+
+RUN mv SEED/* pretrained/ && rm -rf SEED
+
+RUN chmod +x start.sh
+
+CMD ["./start.sh"]

start.sh

@@ -0,0 +1,6 @@
+#!/bin/bash
+
+screen -S script1 -dm bash -c 'python3 gradio_demo/seed_llama_flask.py --image_transform configs/transform/clip_transform.yaml --tokenizer configs/tokenizer/seed_llama_tokenizer.yaml --model configs/llm/seed_llama_14b.yaml   --port 7890 --llm_device cuda:0 --tokenizer_device cuda:0 --offload_tokenizer; exec bash'
+screen -S script2 -dm bash -c 'python3 gradio_demo/seed_llama_gradio.py; exec bash'
+
+tail -f /dev/null