mirror of
https://github.com/AvaLovelace1/LegoGPT.git
synced 2026-05-25 10:19:23 -05:00
apun
37 lines
747 B
TOML
37 lines
747 B
TOML
[project]
|
|
name = "legogpt"
|
|
version = "0.1.0"
|
|
description = "Official repository for LegoGPT"
|
|
readme = "README.md"
|
|
authors = [
|
|
{ name = "Ava Pun", email = "apun@andrew.cmu.edu" }
|
|
]
|
|
requires-python = ">=3.13"
|
|
dependencies = [
|
|
"gurobipy>=12.0.1",
|
|
"peft>=0.15.0",
|
|
"torch>=2.6.0",
|
|
"transformers>=4.50.0",
|
|
"trl>=0.16.0",
|
|
"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"]
|
|
|
|
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
[dependency-groups]
|
|
dev = ["pytest>=8.3.5"]
|