Skip to content
Snippets Groups Projects
Unverified Commit 4eaca1a3 authored by Dmitrii Khizbullin's avatar Dmitrii Khizbullin Committed by GitHub
Browse files

Restrict numpy version to 1.x since 2.0 crashes. (#27)

parent e72b325e
No related branches found
No related tags found
No related merge requests found
...@@ -62,8 +62,9 @@ tqdm = "^4.66.1" ...@@ -62,8 +62,9 @@ tqdm = "^4.66.1"
transformers = "^4.36.2" transformers = "^4.36.2"
wikipedia = "^1.4.0" wikipedia = "^1.4.0"
astunparse = "^1.6.3" astunparse = "^1.6.3"
torch = "^2.1.0" torch = ">=2.1.0, <=2.2.2"
async-timeout = "^4.0.3" async-timeout = "^4.0.3"
numpy = "^1.25.2"
[tool.poetry.group.dev.dependencies] [tool.poetry.group.dev.dependencies]
pytest = "^8.0.2" pytest = "^8.0.2"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment