hufr/pyproject.toml at main · robertsonwang/hufr · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
[tool.poetry]
name = "hufr"
version = "2.0.1"
description = "Redact Text with HuggingFace Models"
authors = ["Robertson Wang <robertsonwang@gmail.com>"]
license = "Apache 2.0"
readme = "README.md"
homepage = "https://github.com/robertsonwang/hufr"
repository = "https://github.com/robertsonwang/hufr"
keywords = ["huggingface", "pii", "ner", "ONNX", "NLP", "redactions"]

[tool.poetry.dependencies]
python = "^3.9"
transformers = "~4.36.2"
torch = "^2.1.2"
# If both ONNX versions are installed, things will not work.
onnxruntime = { version = "^1.13.1", optional = true, markers="extra!='gpu'"}
onnxruntime-gpu = { version = "^1.13.1", optional = true }
onnx = "^1.15.0"

[tool.poetry.extras]
gpu = ["onnxruntime-gpu"]
cpu = ["onnxruntime"]

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"