Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM kernai/refinery-parent-images:v1.23.0-torch-cpu
FROM kernai/refinery-parent-images:v2.0.0-torch-cpu

WORKDIR /program

Expand Down
2 changes: 1 addition & 1 deletion dev.Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM kernai/refinery-parent-images:v1.23.0-torch-cpu
FROM kernai/refinery-parent-images:v2.0.0-torch-cpu

WORKDIR /app

Expand Down
9 changes: 5 additions & 4 deletions gpu-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ blis==0.7.11
# via thinc
boto3==1.39.6
# via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cuda-requirements.txt
botocore==1.39.6
botocore==1.39.10
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cuda-requirements.txt
# boto3
Expand Down Expand Up @@ -73,7 +73,7 @@ cymem==2.0.11
# preshed
# spacy
# thinc
fastapi==0.115.2
fastapi==0.116.1
# via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cuda-requirements.txt
filelock==3.18.0
# via
Expand Down Expand Up @@ -290,7 +290,7 @@ requests==2.32.4
# weasel
rich==14.0.0
# via typer
s3transfer==0.13.0
s3transfer==0.13.1
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cuda-requirements.txt
# boto3
Expand Down Expand Up @@ -339,7 +339,7 @@ srsly==2.5.1
# spacy
# thinc
# weasel
starlette==0.40.0
starlette==0.47.2
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cuda-requirements.txt
# fastapi
Expand Down Expand Up @@ -393,6 +393,7 @@ typing-extensions==4.14.1
# pydantic
# pydantic-core
# sentence-transformers
# starlette
# torch
# typer
urllib3==2.5.0
Expand Down
2 changes: 1 addition & 1 deletion gpu.Dockerfile
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
FROM kernai/refinery-parent-images:v1.23.0-torch-cuda
FROM kernai/refinery-parent-images:v2.0.0-torch-cuda

WORKDIR /program

Expand Down
9 changes: 5 additions & 4 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ blis==0.7.11
# via thinc
boto3==1.39.6
# via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
botocore==1.39.6
botocore==1.39.10
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
# boto3
Expand Down Expand Up @@ -73,7 +73,7 @@ cymem==2.0.11
# preshed
# spacy
# thinc
fastapi==0.115.2
fastapi==0.116.1
# via -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
filelock==3.18.0
# via
Expand Down Expand Up @@ -253,7 +253,7 @@ requests==2.32.4
# weasel
rich==14.0.0
# via typer
s3transfer==0.13.0
s3transfer==0.13.1
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
# boto3
Expand Down Expand Up @@ -302,7 +302,7 @@ srsly==2.5.1
# spacy
# thinc
# weasel
starlette==0.40.0
starlette==0.47.2
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
# fastapi
Expand Down Expand Up @@ -354,6 +354,7 @@ typing-extensions==4.14.1
# pydantic
# pydantic-core
# sentence-transformers
# starlette
# torch
# typer
urllib3==2.5.0
Expand Down
9 changes: 5 additions & 4 deletions requirements/torch-cpu-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ argon2-cffi-bindings==21.2.0
# via argon2-cffi
boto3==1.39.6
# via -r common-requirements.in
botocore==1.39.6
botocore==1.39.10
# via
# boto3
# s3transfer
Expand All @@ -30,7 +30,7 @@ charset-normalizer==3.4.2
# via requests
click==8.2.1
# via uvicorn
fastapi==0.115.2
fastapi==0.116.1
# via -r mini-requirements.in
filelock==3.18.0
# via
Expand Down Expand Up @@ -117,7 +117,7 @@ requests==2.32.4
# -r mini-requirements.in
# huggingface-hub
# transformers
s3transfer==0.13.0
s3transfer==0.13.1
# via boto3
safetensors==0.5.3
# via transformers
Expand All @@ -137,7 +137,7 @@ sniffio==1.3.1
# via anyio
sqlalchemy==1.4.42
# via -r common-requirements.in
starlette==0.40.0
starlette==0.47.2
# via fastapi
sympy==1.14.0
# via torch
Expand All @@ -161,6 +161,7 @@ typing-extensions==4.14.1
# minio
# pydantic
# pydantic-core
# starlette
# torch
urllib3==2.5.0
# via
Expand Down
9 changes: 5 additions & 4 deletions requirements/torch-cuda-requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ argon2-cffi-bindings==21.2.0
# via argon2-cffi
boto3==1.39.6
# via -r common-requirements.in
botocore==1.39.6
botocore==1.39.10
# via
# boto3
# s3transfer
Expand All @@ -30,7 +30,7 @@ charset-normalizer==3.4.2
# via requests
click==8.2.1
# via uvicorn
fastapi==0.115.2
fastapi==0.116.1
# via -r mini-requirements.in
filelock==3.18.0
# via
Expand Down Expand Up @@ -117,7 +117,7 @@ requests==2.32.4
# -r mini-requirements.in
# huggingface-hub
# transformers
s3transfer==0.13.0
s3transfer==0.13.1
# via boto3
safetensors==0.5.3
# via transformers
Expand All @@ -137,7 +137,7 @@ sniffio==1.3.1
# via anyio
sqlalchemy==1.4.42
# via -r common-requirements.in
starlette==0.40.0
starlette==0.47.2
# via fastapi
sympy==1.14.0
# via torch
Expand All @@ -161,6 +161,7 @@ typing-extensions==4.14.1
# minio
# pydantic
# pydantic-core
# starlette
# torch
urllib3==2.5.0
# via
Expand Down