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
|