diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/pyproject.toml b/pyproject.toml index 43943e7..43ab115 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -35,6 +35,10 @@ wandb = "^0.9.6" torch-summary = "^1.4.2" 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] pytest = "^5.4.2" |