mknolan commited on
Commit
8dc8d0b
·
verified ·
1 Parent(s): 57d5e90

Upload InternVL2 implementation

Browse files
Files changed (2) hide show
  1. Dockerfile +8 -7
  2. 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 careful order
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 acceleration libraries
42
- pip3 install --no-cache-dir accelerate==0.27.2 bitsandbytes==0.41.3 && \
 
 
 
43
  # Install gradio
44
  pip3 install --no-cache-dir gradio==3.38.0 && \
45
- # Install remaining requirements
46
- pip3 install --no-cache-dir -r requirements.txt && \
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.27.2
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