updates
Browse files- Dockerfile +2 -1
- requirements.txt +7 -0
Dockerfile
CHANGED
|
@@ -5,7 +5,8 @@ FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu22.04
|
|
| 5 |
# and install (with the yes flag `-y`)
|
| 6 |
# Python and Pip
|
| 7 |
RUN apt-get update && apt-get install -y \
|
| 8 |
-
python3-pip
|
|
|
|
| 9 |
|
| 10 |
|
| 11 |
WORKDIR /code
|
|
|
|
| 5 |
# and install (with the yes flag `-y`)
|
| 6 |
# Python and Pip
|
| 7 |
RUN apt-get update && apt-get install -y \
|
| 8 |
+
python3-pip \
|
| 9 |
+
unzip
|
| 10 |
|
| 11 |
|
| 12 |
WORKDIR /code
|
requirements.txt
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
fastapi==0.74.*
|
| 2 |
+
requests==2.27.*
|
| 3 |
+
uvicorn[standard]==0.17.*
|
| 4 |
+
transformers
|
| 5 |
+
torch
|
| 6 |
+
diffusers
|
| 7 |
+
accelerate
|