Upload InternVL2 implementation
Browse files- Dockerfile +8 -7
- requirements.txt +1 -1
Dockerfile
CHANGED
|
@@ -30,7 +30,7 @@ WORKDIR /app
|
|
| 30 |
# Copy requirements file
|
| 31 |
COPY requirements.txt .
|
| 32 |
|
| 33 |
-
# Upgrade pip and install dependencies in
|
| 34 |
RUN pip3 install --no-cache-dir --upgrade pip && \
|
| 35 |
# Install torch and torchvision first
|
| 36 |
pip3 install --no-cache-dir torch==2.0.1 torchvision==0.15.2 && \
|
|
@@ -38,14 +38,15 @@ RUN pip3 install --no-cache-dir --upgrade pip && \
|
|
| 38 |
pip3 install --no-cache-dir numpy==1.24.3 scipy==1.11.3 requests==2.31.0 && \
|
| 39 |
# Install huggingface dependencies
|
| 40 |
pip3 install --no-cache-dir transformers==4.37.2 safetensors==0.4.1 huggingface_hub==0.19.4 && \
|
| 41 |
-
# Install
|
| 42 |
-
pip3 install --no-cache-dir accelerate==0.
|
|
|
|
|
|
|
|
|
|
| 43 |
# Install gradio
|
| 44 |
pip3 install --no-cache-dir gradio==3.38.0 && \
|
| 45 |
-
# Install remaining requirements
|
| 46 |
-
pip3 install --no-cache-dir
|
| 47 |
-
# Install specific version of lmdeploy that's known to work
|
| 48 |
-
pip3 install --no-cache-dir "lmdeploy>=0.5.0,<0.6.0"
|
| 49 |
|
| 50 |
# Copy the application files
|
| 51 |
COPY . .
|
|
|
|
| 30 |
# Copy requirements file
|
| 31 |
COPY requirements.txt .
|
| 32 |
|
| 33 |
+
# Upgrade pip and install dependencies in specific order to avoid conflicts
|
| 34 |
RUN pip3 install --no-cache-dir --upgrade pip && \
|
| 35 |
# Install torch and torchvision first
|
| 36 |
pip3 install --no-cache-dir torch==2.0.1 torchvision==0.15.2 && \
|
|
|
|
| 38 |
pip3 install --no-cache-dir numpy==1.24.3 scipy==1.11.3 requests==2.31.0 && \
|
| 39 |
# Install huggingface dependencies
|
| 40 |
pip3 install --no-cache-dir transformers==4.37.2 safetensors==0.4.1 huggingface_hub==0.19.4 && \
|
| 41 |
+
# Install lmdeploy and its dependencies first
|
| 42 |
+
pip3 install --no-cache-dir "accelerate==0.30.0" && \
|
| 43 |
+
pip3 install --no-cache-dir "lmdeploy==0.5.3" && \
|
| 44 |
+
# Install other acceleration libraries
|
| 45 |
+
pip3 install --no-cache-dir bitsandbytes==0.41.3 && \
|
| 46 |
# Install gradio
|
| 47 |
pip3 install --no-cache-dir gradio==3.38.0 && \
|
| 48 |
+
# Install any remaining requirements
|
| 49 |
+
pip3 install --no-cache-dir packaging==23.2 pyyaml==6.0.1 tqdm==4.66.1 typing-extensions==4.8.0 openai==1.6.1
|
|
|
|
|
|
|
| 50 |
|
| 51 |
# Copy the application files
|
| 52 |
COPY . .
|
requirements.txt
CHANGED
|
@@ -7,7 +7,7 @@ numpy==1.24.3
|
|
| 7 |
scipy==1.11.3
|
| 8 |
gradio==3.38.0
|
| 9 |
requests==2.31.0
|
| 10 |
-
accelerate==0.
|
| 11 |
bitsandbytes==0.41.3
|
| 12 |
safetensors==0.4.1
|
| 13 |
openai==1.6.1
|
|
|
|
| 7 |
scipy==1.11.3
|
| 8 |
gradio==3.38.0
|
| 9 |
requests==2.31.0
|
| 10 |
+
accelerate==0.30.0
|
| 11 |
bitsandbytes==0.41.3
|
| 12 |
safetensors==0.4.1
|
| 13 |
openai==1.6.1
|