Skip to content

Commit 0040bab

Browse files
committed
add torch dependency for CPU support in requirements
1 parent 87cb744 commit 0040bab

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,4 +26,5 @@ flaml
2626
# notebooks
2727
ipykernel
2828
pyreadr
29+
torch==2.1.2+cpu --extra-index-url https://download.pytorch.org/whl/cpu
2930
tabpfn

0 commit comments

Comments
 (0)