Update docs requirements

hmorimitsu · Dec 4, 2024 · 4187813 · 4187813
1 parent 069b237
commit 4187813
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,6 +1,22 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+
 numpydoc==1.8.0
 sphinx==8.1.3
 sphinx_rtd_theme==3.0.2
 
-ptlflow
-timm==1.0.11
+torch==2.5.1
+torchvision==0.20.1
+
+einops<0.9
+h5py<3.13
+kaleido<0.3
+lightning[pytorch-extra]>=2.1,<2.5
+loguru<0.8
+opencv-python<4.11
+pandas<2.3
+plotly<5.25
+pypng==0.20220715.0
+scipy<1.15
+tabulate<0.10
+tensorboard<2.19
+timm<1.1