diff options
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/pyproject.toml b/pyproject.toml index 252678f..abf69ed 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -39,6 +39,8 @@ sentencepiece = "^0.1.95" pytorch-lightning = "^1.2.4" Pillow = "^8.1.2" madgrad = "^1.0" +editdistance = "^0.5.3" +torchmetrics = "^0.2.0" [tool.poetry.dev-dependencies] pytest = "^5.4.2" |