From 63097449f171f7eeae12a750114bf276fedd7056 Mon Sep 17 00:00:00 2001
From: octadion <dyonoct@gmail.com>
Date: Tue, 27 Feb 2024 15:55:59 +0700
Subject: [PATCH] fix: curl download model yolo nas

---
 Dockerfile | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 3668254..c307e35 100644
--- 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