diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/pyproject.toml b/pyproject.toml index 95e32a6..c977270 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -22,8 +22,8 @@ sphinx_rtd_theme = "^0.4.3" boltons = "^20.1.0" h5py = "^2.10.0" toml = "^0.10.1" -torch = "^1.6.0" -torchvision = "^0.7.0" +torch = "^1.7.0" +torchvision = "^0.8.1" loguru = "^0.5.0" matplotlib = "^3.2.1" tqdm = "^4.46.1" @@ -37,7 +37,6 @@ python-Levenshtein = "^0.12.0" defusedxml = "^0.6.0" pytorch-block-sparse = "^0.1.2" pytorch-metric-learning = "^0.9.92" -faiss-gpu = "^1.6.3" omegaconf = "^2.0.2" [tool.poetry.dev-dependencies] |