File size: 2,847 Bytes
131da64 51303d2 131da64 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 |
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[tool.hatch.build]
sources = ["unidisc"]
[tool.uv]
package = false
default-groups = ["dev", "misc"]
no-build-isolation-package = ["flash-attn"]
[project]
name = "unidisc"
version = "0.0.1"
authors = [{ name="Alexander Swerdlow", email="[email protected]" }]
readme = "README.md"
requires-python = ">=3.10,<3.13"
dependencies = [
"setuptools>=75.8.0",
"torch>=2.6.0",
"torchvision>=0.21.0",
"diffusers~=0.32.2",
"transformers~=4.49.0",
"datasets~=3.2.0",
"numpy~=2.2",
"tensordict~=0.7.2",
"accelerate~=1.5.2",
"lightning_utilities~=0.12.0",
"hydra-core~=1.3.2",
"omegaconf~=2.3.0",
"torchtnt~=0.2.4",
"jaxtyping~=0.2.37",
"einops~=0.8.0",
"timm~=1.0.15",
"wandb~=0.19.6",
"image_utilities==0.0.3*",
"typer~=0.15.1",
"torchmetrics==1.6.1",
"rich~=13.9.4",
"fsspec",
"pandas",
"ml_collections",
"scikit-learn",
"torchinfo",
"sentencepiece",
"hf_transfer",
"ipdb",
"ipython",
"lovely-tensors",
]
[dependency-groups]
dev = [
"peft",
"braceexpand",
"h5py",
"pynvml",
"evaluate",
"mauve-text",
"clean-fid",
"hpsv2x==1.2.0",
"open_clip_torch",
"T2IBenchmark",
"clip",
"python-fasthtml~=0.12.1",
"MonsterUI~=0.0.34",
"fastapi~=0.115.8",
"flash-attn~=2.7.4",
]
misc = [
"flask",
"werkzeug",
"sentence_transformers",
"opencv-python",
"lpips",
"simple_slurm",
"ftfy",
"bitsandbytes",
"requests",
"deepspeed",
]
# Not important and can be difficult to install.
# This may be easier: `uv pip install fairseq --no-deps`
# hard = [
# "fairseq",
# "langchain~=0.3.17",
# "langchain_core~=0.3.15",
# "langchain_groq~=0.2.1",
# ]
[tool.uv.sources]
torch = [{ index = "pytorch-cu124", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },]
torchvision = [{ index = "pytorch-cu124", marker = "sys_platform == 'linux' or sys_platform == 'win32'" },]
webdataset = { git = "https://github.com/alexanderswerdlow/webdataset.git", rev = "67d1d487dc1a9aa6aaf81e6712deaec29c1ae3d3" }
submitit = { git = "https://github.com/alexanderswerdlow/submitit.git", rev = "eb6368c068a9a64e9f09c9128b47c39a81add324" }
T2IBenchmark = { git = "https://github.com/boomb0om/text2image-benchmark.git", rev = "532229f679d7e97ecba61914db7276f95733e707" }
clip = { git = "https://github.com/openai/CLIP.git", rev = "dcba3cb2e2827b402d2701e7e1c7d9fed8a20ef1" }
# hydra-core = { git = "https://github.com/alexanderswerdlow/hydra.git", branch = "working_ci" }
# hydra-submitit-launcher = { git = "https://github.com/alexanderswerdlow/hydra.git", branch = "working_ci", subdirectory = "plugins/hydra_submitit_launcher" }
[[tool.uv.index]]
name = "pytorch-cu124"
url = "https://download.pytorch.org/whl/cu124"
explicit = true
[tool.black]
line-length = 150
target-version = ['py310']
|