new parent image (#30)

* new parent image

* Adds gpu requirements file

---------

Co-authored-by: JWittmeyer <[email protected]>

Author: JWittmeyer

gpu-requirements.txt

@@ -1,6 +1,6 @@
 #
-# This file is autogenerated by pip-compile with python 3.9
-# To update, run:
+# This file is autogenerated by pip-compile with Python 3.9
+# by the following command:
 #
 #    pip-compile --output-file=gpu-requirements.txt requirements/gpu-requirements.in
 #
@@ -48,18 +48,22 @@ cymem==2.0.7
     #   thinc
 embedders==0.0.18
     # via -r requirements/gpu-requirements.in
-fastapi==0.85.1
+fastapi==0.92.0
     # via -r requirements/torch-cuda-requirements.txt
-filelock==3.8.2
+filelock==3.9.0
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   huggingface-hub
     #   transformers
+greenlet==2.0.2
+    # via
+    #   -r requirements/torch-cuda-requirements.txt
+    #   sqlalchemy
 h11==0.14.0
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   uvicorn
-huggingface-hub==0.11.1
+huggingface-hub==0.12.1
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   sentence-transformers
@@ -106,7 +110,7 @@ numpy==1.23.4
     #   thinc
     #   torchvision
     #   transformers
-packaging==22.0
+packaging==23.0
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   huggingface-hub
@@ -136,7 +140,7 @@ python-dateutil==2.8.2
     #   -r requirements/torch-cuda-requirements.txt
     #   botocore
     #   pandas
-pytz==2022.6
+pytz==2022.7.1
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   pandas
@@ -198,7 +202,7 @@ srsly==2.4.5
     #   confection
     #   spacy
     #   thinc
-starlette==0.20.4
+starlette==0.25.0
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   fastapi
@@ -238,15 +242,15 @@ typer==0.4.2
     # via
     #   pathy
     #   spacy
-typing-extensions==4.4.0
+typing-extensions==4.5.0
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   huggingface-hub
     #   pydantic
     #   starlette
     #   torch
     #   torchvision
-urllib3==1.26.13
+urllib3==1.26.14
     # via
     #   -r requirements/torch-cuda-requirements.txt
     #   botocore

gpu.Dockerfile

@@ -1,4 +1,4 @@
-FROM kernai/refinery-parent-images:v1.8.0-torch-cuda
+FROM kernai/refinery-parent-images:v1.8.1-torch-cuda
 
 WORKDIR /program
 

requirements/torch-cuda-requirements.txt

@@ -1,15 +1,15 @@
 #
-# This file is autogenerated by pip-compile with python 3.9
-# To update, run:
+# This file is autogenerated by pip-compile with Python 3.9
+# by the following command:
 #
-#    pip-compile requirements/torch-cuda-requirements.in
+#    pip-compile torch-cuda-requirements.in
 #
 --extra-index-url https://download.pytorch.org/whl/cu113
 
 anyio==3.6.2
     # via starlette
 boto3==1.25.0
-    # via -r requirements/common-requirements.in
+    # via -r common-requirements.in
 botocore==1.28.5
     # via
     #   boto3
@@ -22,15 +22,17 @@ charset-normalizer==2.1.1
     # via requests
 click==8.1.3
     # via uvicorn
-fastapi==0.85.1
-    # via -r requirements/mini-requirements.in
-filelock==3.8.2
+fastapi==0.92.0
+    # via -r mini-requirements.in
+filelock==3.9.0
     # via
     #   huggingface-hub
     #   transformers
+greenlet==2.0.2
+    # via sqlalchemy
 h11==0.14.0
     # via uvicorn
-huggingface-hub==0.11.1
+huggingface-hub==0.12.1
     # via transformers
 idna==3.4
     # via
@@ -41,29 +43,29 @@ jmespath==1.0.1
     #   boto3
     #   botocore
 minio==7.1.12
-    # via -r requirements/common-requirements.in
+    # via -r common-requirements.in
 numpy==1.23.4
     # via
-    #   -r requirements/common-requirements.in
+    #   -r common-requirements.in
     #   pandas
     #   transformers
-packaging==22.0
+packaging==23.0
     # via
     #   huggingface-hub
     #   transformers
 pandas==1.5.1
-    # via -r requirements/common-requirements.in
+    # via -r common-requirements.in
 psycopg2-binary==2.9.4
-    # via -r requirements/common-requirements.in
+    # via -r common-requirements.in
 pydantic==1.10.2
     # via
-    #   -r requirements/mini-requirements.in
+    #   -r mini-requirements.in
     #   fastapi
 python-dateutil==2.8.2
     # via
     #   botocore
     #   pandas
-pytz==2022.6
+pytz==2022.7.1
     # via pandas
 pyyaml==6.0
     # via
@@ -73,7 +75,7 @@ regex==2022.10.31
     # via transformers
 requests==2.28.1
     # via
-    #   -r requirements/mini-requirements.in
+    #   -r mini-requirements.in
     #   huggingface-hub
     #   transformers
 s3transfer==0.6.0
@@ -83,29 +85,29 @@ six==1.16.0
 sniffio==1.3.0
     # via anyio
 sqlalchemy==1.4.42
-    # via -r requirements/common-requirements.in
-starlette==0.20.4
+    # via -r common-requirements.in
+starlette==0.25.0
     # via fastapi
 tokenizers==0.13.2
     # via transformers
 torch==1.13.1
-    # via -r requirements/torch-cuda-requirements.in
+    # via -r torch-cuda-requirements.in
 tqdm==4.64.1
     # via
     #   huggingface-hub
     #   transformers
 transformers==4.23.1
-    # via -r requirements/torch-cuda-requirements.in
-typing-extensions==4.4.0
+    # via -r torch-cuda-requirements.in
+typing-extensions==4.5.0
     # via
     #   huggingface-hub
     #   pydantic
     #   starlette
     #   torch
-urllib3==1.26.13
+urllib3==1.26.14
     # via
     #   botocore
     #   minio
     #   requests
 uvicorn==0.19.0
-    # via -r requirements/mini-requirements.in
+    # via -r mini-requirements.in