From 4187813ce9e42c405f860f27604fac2ae88f5381 Mon Sep 17 00:00:00 2001
From: Henrique Morimitsu <henriquem87@gmail.com>
Date: Wed, 4 Dec 2024 13:03:25 +0800
Subject: [PATCH] Update docs requirements

---
 docs/requirements.txt | 20 ++++++++++++++++++--
 1 file changed, 18 insertions(+), 2 deletions(-)

diff --git a/docs/requirements.txt b/docs/requirements.txt
index 7a0cedf..79b6fd0 100644
--- 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
\ No newline at end of file
+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
\ No newline at end of file