修复torch和numpy版本过新导致整合包无法运行的bug

chn-lee-yumi · Aug 16, 2024 · 7efd868 · 7efd868
1 parent 0e81228
commit 7efd868
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
-numpy>=1.20.3
+numpy>=1.20.3,<2.0
 Flask>=2.2.2
 SQLAlchemy>=2.0.20
-torch>=1.13.1
+torch>=2.0,<2.4
 Pillow>=8.1.0
 pillow-heif>=0.14.0
 transformers>=4.28.1