pyproject.toml

[project]
name = "unique3d"
version = "0.1.0"
description = "Add your description here"
readme = "README.md"
requires-python = ">=3.10"
dependencies = [
    "accelerate>=1.10.1",
    "datasets>=4.0.0",
    "diffusers==0.33.1",
    "fire>=0.7.1",
    "gradio>=5.44.1",
    "hf-transfer>=0.1.9",
    "jaxtyping>=0.3.2",
    "numpy<2",
    "omegaconf>=2.3.0",
    "opencv-python-headless>=4.11.0.86",
    "peft>=0.17.1",
    "pygltflib>=1.16.5",
    "pymeshlab==2023.12.post3",
    "pytorch3d",
    "rembg[gpu]==2.0.62",
    "spaces>=0.40.1",
    "torch==2.2.0",
    "torch-scatter",
    "transformers==4.48.2",
    "trimesh>=4.7.4",
    "typeguard>=4.4.4",
    "wandb>=0.21.3",
    "xformers==0.0.24",
]

[tool.uv.sources]
pytorch3d = { url = "https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl" }
torch-scatter = { url = "https://data.pyg.org/whl/torch-2.2.0%2Bcu121/torch_scatter-2.1.2%2Bpt22cu121-cp310-cp310-linux_x86_64.whl" }