- update requirements

cybercongress · May 18, 2024 · 15d645c · 15d645c
1 parent 64c0bbe
commit 15d645c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/requirements/dev.txt b/requirements/dev.txt
@@ -9,4 +9,5 @@ ddt==1.6.0
 hypothesis==6.81.1
 flake8==7.0.0
 mypy==1.8.0
-types-retry==0.9.9.4
+types-retry==0.9.9.4
+jupyter>=1.0.0
diff --git a/requirements/prod.txt b/requirements/prod.txt
@@ -29,7 +29,8 @@ rich
 setuptools<=65.6.3
 shtab==1.6.5
 termcolor
-torch>=1.13.1
+torch>=1.13.1+cpu
+torchvision>=0.13.1+cpu
 tqdm
 uvicorn==0.22.0
 wheel