[python] Fix torch environment issue.

buddy-compiler · Apr 10, 2024 · 5f7f69c · 5f7f69c
1 parent d8dc1c2
commit 5f7f69c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,6 @@
 --pre --extra-index-url https://download.pytorch.org/whl/cpu
 torch == 2.1.2
+numpy < 2
 transformers == 4.33.1
 tokenizers == 0.13.3
 sentencepiece == 0.1.99