fix: curl download model yolo nas

octadion · Feb 27, 2024 · 6309744 · 6309744
1 parent 36de6f3
commit 6309744
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -11,13 +11,10 @@ RUN apt-get update -y && apt-get install -y \
     curl \
     libgl1-mesa-glx \
     libglib2.0-dev \
-    git-lfs \
     && pip install --upgrade pip && pip install -r requirements.txt
 
-RUN git lfs install --force && \
-    git clone -b 'main' --single-branch --depth 1 https://github.com/octadion/visionllm.git && \
-    cd visionllm && \
-    git lfs pull
+RUN mkdir -p models/yolo-nas
+RUN curl -L https://github.com/octadion/visionllm/raw/main/models/yolo-nas/ckpt_best2.pth -o models/yolo-nas/ckpt_best2.pth
 
 RUN mkdir -p models/llm
 RUN curl -L https://huggingface.co/TheBloke/zephyr-7B-beta-GGUF/resolve/main/zephyr-7b-beta.Q4_K_S.gguf -o models/llm/zephyr-7b-beta.Q4_K_S.gguf