fix depend
Browse files- Dockerfile +1 -1
- requirements.txt +3 -1
Dockerfile
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
# Use a base image with Python
|
| 2 |
-
|
| 3 |
|
| 4 |
# Install git and other build tools
|
| 5 |
# The specific command depends on the base image's package manager.
|
|
|
|
| 1 |
# Use a base image with Python
|
| 2 |
+
RUN python3 -m pip install --upgrade pip
|
| 3 |
|
| 4 |
# Install git and other build tools
|
| 5 |
# The specific command depends on the base image's package manager.
|
requirements.txt
CHANGED
|
@@ -3,4 +3,6 @@ transformers
|
|
| 3 |
unsloth[cu121-ampere-torch230] @ git+https://github.com/unslothai/unsloth.git@August-2025
|
| 4 |
streamlit
|
| 5 |
fastapi
|
| 6 |
-
uvicorn
|
|
|
|
|
|
|
|
|
| 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
|