bla commited on
Commit
987f5b5
·
verified ·
1 Parent(s): 7843df1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -32,13 +32,13 @@ RUN pip install --no-cache-dir --upgrade pip && \
32
  pydantic \
33
  supervision && \
34
  # Install AutoAWQ last
35
- pip install --no-cache-dir autoawq
36
 
37
 
38
  RUN pip install git+https://github.com/huggingface/transformers accelerate
39
 
40
 
41
- RUN pip install git+https://github.com/intel/intel-extension-for-pytorch.git
42
  # Copy application files
43
  COPY --chown=user:user . /app
44
 
 
32
  pydantic \
33
  supervision && \
34
  # Install AutoAWQ last
35
+ pip install --no-cache-dir autoawq[cpu]
36
 
37
 
38
  RUN pip install git+https://github.com/huggingface/transformers accelerate
39
 
40
 
41
+ RUN pip install intel-extension-for-pytorch
42
  # Copy application files
43
  COPY --chown=user:user . /app
44