Create infer uv script

This commit is contained in:
apun
2025-04-01 09:48:04 -04:00
parent 6a24c14466
commit 2f15434fc9
2 changed files with 2 additions and 52 deletions
+2 -11
View File
@@ -16,17 +16,8 @@ dependencies = [
"wandb>=0.19.8",
]
[project.optional-dependencies]
build = ["torch", "setuptools", "packaging"]
compile = ["flash-attn"]
[tool.uv]
no-build-isolation-package = ["flash-attn"]
[[tool.uv.dependency-metadata]]
name = "flash-attn"
version = "2.6.3"
requires-dist = ["torch", "einops"]
[project.scripts]
infer = "legogpt.infer:main"
[build-system]
requires = ["hatchling"]