chore(packages): bump or lock down more versions, add more packages

gadicc · gadicc · commit 6d6a89e50d49 · 2022-12-09T18:56:07.000Z
diff --git a/Dockerfile b/Dockerfile
@@ -115,9 +115,6 @@ ADD precision.py .
 ARG PIPELINE="ALL"
 ENV PIPELINE=${PIPELINE}
 
-ARG USE_DREAMBOOTH=0
-ENV USE_DREAMBOOTH=${USE_DREAMBOOTH}
-
 ARG AWS_ACCESS_KEY_ID
 ARG AWS_SECRET_ACCESS_KEY
 ARG AWS_DEFAULT_REGION
@@ -163,6 +160,9 @@ ARG USE_PATCHMATCH=0
 RUN if [ "$USE_PATCHMATCH" = "1" ] ; then apt-get install -yqq python3-opencv ; fi
 COPY --from=patchmatch /tmp/PyPatchMatch PyPatchMatch
 
+ARG USE_DREAMBOOTH=0
+ENV USE_DREAMBOOTH=${USE_DREAMBOOTH}
+
 RUN if [ "$USE_DREAMBOOTH" = "1" ] ; then \
     # By specifying the same torch version as conda, it won't download again.
     # Without this, it will upgrade torch, break xformers, make bigger image.
diff --git a/requirements.txt b/requirements.txt
@@ -5,11 +5,19 @@ sanic==22.6.2
 transformers==4.22.2
 scipy==1.9.3
 requests_futures==1.0.0
-numpy==1.23.4
+# numpy==1.23.4
+numpy==1.23.5
 scikit-image==0.19.3
-accelerate==0.13.2
-triton==2.0.0.dev20221105
+# accelerate==0.13.2
+accelerate==0.15.0
+# triton==2.0.0.dev20221105
+triton==2.0.0.dev20221202
 ftfy==6.1.1
 spacy==3.4.3
 boto3==1.26.12
-k-diffusion==0.0.12
+k-diffusion==0.0.12
+safetensors==0.2.5
+# must match conda installed version in Dockerfile
+torch==1.12.1
+# xformers 0.0.15.dev344+git.8910bb5 requires pyre-extensions==0.0.23
+pyre-extensions==0.0.23
