Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -50,7 +50,7 @@ RUN --mount=type=cache,target=/app/.npm \
|
|
| 50 |
|
| 51 |
RUN npm run build
|
| 52 |
|
| 53 |
-
FROM ghcr.io/huggingface/text-generation-inference:
|
| 54 |
|
| 55 |
ARG MODEL_NAME
|
| 56 |
ARG MODEL_PARAMS
|
|
|
|
| 50 |
|
| 51 |
RUN npm run build
|
| 52 |
|
| 53 |
+
FROM ghcr.io/huggingface/text-generation-inference:2.1.1
|
| 54 |
|
| 55 |
ARG MODEL_NAME
|
| 56 |
ARG MODEL_PARAMS
|