Hugging Face's logo

Spaces:
nroggendorff
/
create-caption
Paused

App Files Community
nroggendorff commited on
Commit
55c4eff
·
verified ·
1 Parent(s): df9c093

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -10,7 +10,7 @@ RUN python3 -m pip install --upgrade pip
10
  WORKDIR /app
11
  COPY . .
12
 
13
- RUN pip install --no-cache-dir torch==2.4.0+cu123 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu121
14
 
15
  RUN pip install --no-cache-dir \
16
  "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.3.11/flash_attn-2.8.0+cu121torch2.4-cp310-cp310-linux_x86_64.whl"
 
10
  WORKDIR /app
11
  COPY . .
12
 
13
+ RUN pip install --no-cache-dir torch==2.4.0+cu121 torchvision==0.19.0 --extra-index-url https://download.pytorch.org/whl/cu121
14
 
15
  RUN pip install --no-cache-dir \
16
  "https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.3.11/flash_attn-2.8.0+cu121torch2.4-cp310-cp310-linux_x86_64.whl"