add script entrypoint

benjaminye · benjaminye · commit 2c438d083efb · 2024-04-02T23:20:19.000-04:00
diff --git a/pyproject.toml b/pyproject.toml
@@ -7,6 +7,9 @@ license = "Apache 2.0"
 readme = "README.md"
 packages = [{include = "llmtune"}]
 
+[tool.poetry.scripts]
+llmtune = "llmtune.cli.toolkit:cli"
+
 [tool.poetry.dependencies]
 python = ">=3.9, <=3.12"
 transformers = "~4.37.2"
