adding whl, updating deps

.gitattributes

@@ -1,37 +0,0 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
-checkpoints/ filter=lfs diff=lfs merge=lfs -text
-checkpoints/depth_pro.pt filter=lfs diff=lfs merge=lfs -text

Dockerfile

@@ -1,61 +0,0 @@
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu20.04
-
-ARG DEBIAN_FRONTEND=noninteractive
-ENV CUDA_HOME /usr/local/cuda-11.8/
-
-WORKDIR /app
-
-ENV PATH="/usr/local/cuda-11.8/bin:${PATH}"
-ENV LD_LIBRARY_PATH="/usr/local/cuda-11.8/lib64:${LD_LIBRARY_PATH}"
-
-RUN apt-get update && apt-get install -y software-properties-common wget && \
-    add-apt-repository ppa:deadsnakes/ppa && \
-    apt-get update && \
-    apt-get install -y build-essential git wget curl && \
-    apt-get install -y python3.10 python3.10-dev python3.10-distutils python3-venv && \
-    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1 && \
-    update-alternatives --set python3 /usr/bin/python3.10 && \
-    apt-get install -y zlib1g-dev libexpat1-dev
-
-RUN wget https://github.com/Kitware/CMake/releases/download/v3.26.4/cmake-3.26.4-linux-x86_64.sh && \
-    chmod +x cmake-3.26.4-linux-x86_64.sh && \
-    ./cmake-3.26.4-linux-x86_64.sh --skip-license --prefix=/usr/local && \
-    rm cmake-3.26.4-linux-x86_64.sh
-
-RUN wget https://bootstrap.pypa.io/get-pip.py && \
-    python3 get-pip.py && \
-    rm get-pip.py
-
-RUN python3 -m pip install --upgrade pip && python3 -m pip install setuptools==65.0.1 wheel spacy==3.7.5
-RUN python3 -m spacy download en_core_web_sm
-
-RUN python3 -m pip install numpy==1.21.0
-RUN python3 -m pip install scikit-learn==1.0.2 --prefer-binary
-
-RUN apt-get install --no-install-recommends wget ffmpeg=7:* \
-    libsm6=2:* libxext6=2:* git=1:* vim=2:* -y \
-    && apt-get clean && apt-get autoremove && rm -rf /var/lib/apt/lists/*
-
-RUN wget https://github.com/mikefarah/yq/releases/download/v4.30.8/yq_linux_amd64 -O /usr/bin/yq \
-    && chmod +x /usr/bin/yq
-
-RUN pip install git+https://github.com/apple/ml-depth-pro.git
-RUN pip install 'git+https://github.com/facebookresearch/sam2.git'
-RUN pip install git+https://github.com/openai/CLIP.git
-
-RUN pip install --upgrade torch==2.4.0+cu118 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu118
-
-COPY . /app
-RUN pip install -r requirements.txt
-RUN pip uninstall -y flash_attn
-RUN pip install git+https://github.com/Dao-AILab/[email protected]
-
-RUN pip uninstall -y onnxruntime onnxruntime-gpu
-RUN pip install onnxruntime-gpu==1.18.1
-
-# Expose the port Gradio will run on
-EXPOSE 7860
-
-# Run the Gradio app
-CMD ["python3", "app.py"]
-

extra_deps/flash_attn-2.7.0.post2-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb4eb73682931a7c591a97ff12926b392ebea4c67e8d9846537925fbfcddaecf
+size 183278540

requirements.txt

@@ -18,6 +18,7 @@ onnxsim==0.4.35
 scipy==1.12.0
 litellm==1.25.2
 pycocotools==2.0.6
+onnxruntime-gpu
 pandas
 html5lib
 datasets
@@ -25,7 +26,4 @@ datasets
 git+https://github.com/apple/ml-depth-pro.git
 git+https://github.com/facebookresearch/sam2.git
 git+https://github.com/openai/CLIP.git
-
-git+https://github.com/Dao-AILab/[email protected]
-
-onnxruntime-gpu
+./extra_deps/flash_attn-2.7.0.post2-cp310-cp310-linux_x86_64.whl