jisaacso219 commited on
Commit
654dc4a
·
verified ·
1 Parent(s): 31663d5

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -1
Dockerfile CHANGED
@@ -16,10 +16,13 @@ WORKDIR /app
16
  # Copy application files
17
  COPY . /app
18
 
19
- # Install Python dependencies
20
  RUN pip install --upgrade pip \
 
 
21
  && pip install -r requirements.txt \
22
  && pip install huggingface_hub
 
23
 
24
  # Download GGUF model at build time into models/
25
  RUN mkdir -p models \
 
16
  # Copy application files
17
  COPY . /app
18
 
19
+ # Install Python dependencies, forcing CPU-only PyTorch
20
  RUN pip install --upgrade pip \
21
+ && pip install --index-url https://download.pytorch.org/whl/cpu \
22
+ torch==2.1.0+cpu torchvision==0.16.0+cpu torchaudio==2.1.0+cpu \
23
  && pip install -r requirements.txt \
24
  && pip install huggingface_hub
25
+ && pip install huggingface_hub
26
 
27
  # Download GGUF model at build time into models/
28
  RUN mkdir -p models \