Skip to content
Merged
Show file tree
Hide file tree
Changes from 6 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
13 changes: 7 additions & 6 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 All @@ -320,12 +320,12 @@ tokenizers==0.21.2
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
# transformers
torch==2.7.1
torch==2.7.1+cpu
# via
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
# sentence-transformers
# torchvision
torchvision==0.22.1
torchvision==0.22.1+cpu
# via -r requirements/requirements.in
tqdm==4.67.1
# via
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