Skip to content

Commit 73497c5

Browse files
committed
Move torch and sentencepiece to dev dependencies
- Also add setuptools_cpp to dev dependencies
1 parent f38222a commit 73497c5

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

pyproject.toml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,12 @@ packages = [{ include = "llamacpp", from = "." }]
1111

1212
[tool.poetry.dependencies]
1313
python = "^3.6"
14+
15+
[tool.poetry.group.dev.dependencies]
16+
# Require torch and sentencepiece for running the convert script
1417
torch = "^1.13.1"
1518
sentencepiece = "^0.1.97"
19+
setuptools-cpp = "^0.1.0"
1620

1721
[build-system]
1822
requires = ["poetry>=0.12", "setuptools", "wheel", "setuptools-cpp"]

0 commit comments

Comments
 (0)