Skip to content

Commit 5eaf369

Browse files
authored
bump torch version (#21)
* new parent image * fixes parent image version number * sets correct requirements
1 parent cd6ba8f commit 5eaf369

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM kernai/refinery-parent-images:v1.7.0-torch-cpu
1+
FROM kernai/refinery-parent-images:v1.8.0-torch-cpu
22

33
RUN apt-get update && \
44
apt-get install --no-install-recommends -y curl && \

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,7 @@ tokenizers==0.13.2
138138
# via
139139
# -r requirements/torch-cpu-requirements.txt
140140
# transformers
141-
torch==1.12.1
141+
torch==1.13.1
142142
# via
143143
# -r requirements/torch-cpu-requirements.txt
144144
# sequencelearn

requirements/torch-cpu-requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ starlette==0.20.4
8888
# via fastapi
8989
tokenizers==0.13.2
9090
# via transformers
91-
torch==1.12.1
91+
torch==1.13.1
9292
# via -r requirements/torch-cpu-requirements.in
9393
tqdm==4.64.1
9494
# via

0 commit comments

Comments
 (0)