peihsin0715
commited on
Commit
·
d02be95
1
Parent(s):
e8cf564
Fix model loading: install safetensors (and accelerate) for openai-community/gpt2
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -22,6 +22,7 @@ ENV PATH="/opt/venv/bin:${PATH}"
|
|
| 22 |
RUN python -m pip install --upgrade pip setuptools wheel
|
| 23 |
|
| 24 |
RUN pip install --index-url https://download.pytorch.org/whl/cpu "torch==2.3.1"
|
|
|
|
| 25 |
|
| 26 |
COPY backend/requirements.txt ./backend/requirements.txt
|
| 27 |
RUN sed -i 's/^[Tt]orch[[:space:]=<>!].*/# torch pinned separately (CPU)/' backend/requirements.txt || true
|
|
|
|
| 22 |
RUN python -m pip install --upgrade pip setuptools wheel
|
| 23 |
|
| 24 |
RUN pip install --index-url https://download.pytorch.org/whl/cpu "torch==2.3.1"
|
| 25 |
+
RUN pip install --no-cache-dir safetensors accelerate
|
| 26 |
|
| 27 |
COPY backend/requirements.txt ./backend/requirements.txt
|
| 28 |
RUN sed -i 's/^[Tt]orch[[:space:]=<>!].*/# torch pinned separately (CPU)/' backend/requirements.txt || true
|