Skip to content

Commit 3d6ef01

Browse files
Hotfix embedders (#154)
* fix: embedders imports * fix: requirements +cpu and gpu.Dockerfile
1 parent 61f5756 commit 3d6ef01

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

gpu.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ WORKDIR /program
44

55
COPY gpu-requirements.txt .
66

7-
RUN python3.9 -m pip install --no-cache-dir -r gpu-requirements.txt
7+
RUN pip3 install --no-cache-dir -r gpu-requirements.txt
88

99
COPY / .
1010

requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -320,12 +320,12 @@ tokenizers==0.21.2
320320
# via
321321
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
322322
# transformers
323-
torch==2.7.1+cpu
323+
torch==2.7.1
324324
# via
325325
# -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
326326
# sentence-transformers
327327
# torchvision
328-
torchvision==0.22.1+cpu
328+
torchvision==0.22.1
329329
# via -r requirements/requirements.in
330330
tqdm==4.67.1
331331
# via

0 commit comments

Comments
 (0)