added fixed
Browse files- Dockerfile +1 -0
- requirements.txt +2 -2
Dockerfile
CHANGED
|
@@ -5,6 +5,7 @@ FROM python:3.10-slim
|
|
| 5 |
# The specific command depends on the base image's package manager.
|
| 6 |
# For Debian/Ubuntu-based images like python:3.10-slim, it's apt-get.
|
| 7 |
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
|
|
|
|
| 8 |
|
| 9 |
# Set the working directory in the container
|
| 10 |
WORKDIR /app
|
|
|
|
| 5 |
# The specific command depends on the base image's package manager.
|
| 6 |
# For Debian/Ubuntu-based images like python:3.10-slim, it's apt-get.
|
| 7 |
RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
|
| 8 |
+
RUN python3 -m pip install --upgrade pip
|
| 9 |
|
| 10 |
# Set the working directory in the container
|
| 11 |
WORKDIR /app
|
requirements.txt
CHANGED
|
@@ -1,8 +1,8 @@
|
|
|
|
|
|
|
|
| 1 |
torch
|
| 2 |
transformers
|
| 3 |
unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git@August-2025
|
| 4 |
streamlit
|
| 5 |
fastapi
|
| 6 |
uvicorn
|
| 7 |
-
packaging
|
| 8 |
-
flash-attn>=2.6.3
|
|
|
|
| 1 |
+
packaging
|
| 2 |
+
flash-attn>=2.6.3
|
| 3 |
torch
|
| 4 |
transformers
|
| 5 |
unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git@August-2025
|
| 6 |
streamlit
|
| 7 |
fastapi
|
| 8 |
uvicorn
|
|
|
|
|
|