pyproject.toml

[tool.poetry]
name = "mlserver-huggingface"
version = "1.5.0.dev1"
description = "HuggingFace runtime for MLServer"
authors = ["Seldon Technologies Ltd. <hello@seldon.io>"]
license = "Apache-2.0"
readme = "README.md"
packages = [{include = "mlserver_huggingface"}]

[tool.poetry.dependencies]
python = ">=3.9,<3.12"
mlserver = "*"
tensorflow = "*"
pillow = "*"
tf-keras = "<3.0"
optimum = {extras = ["onnxruntime"], version = ">=1.4,<2.0"}
pydantic = "<2.0.0"
fugashi = {version = "^1.3.0", optional = true}
unidic = {version = "^1.1.0", optional = true}
unidic-lite = {version = "^1.0.8", optional = true}
ipadic = {version = "^1.0.0", optional = true}

[tool.poetry.group.dev.dependencies]
mlserver = {path = "../..", develop = true}
torch = "2.2.2"
transformers = ">=4.30,<5.0"
tensorflow = ">=2.12,<2.17"
tf-keras = "<3.0"

[tool.poetry.extras]
japanese = ["fugashi", "unidic", "unidic-lite", "ipadic"]

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