Update

.python-version

@@ -0,0 +1 @@
+3.10

pyproject.toml

@@ -0,0 +1,35 @@
+[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" }

requirements.txt

@@ -1,26 +1,497 @@
+# This file was autogenerated by uv via the following command:
+#    uv pip compile pyproject.toml -o requirements.txt
+accelerate==1.10.1
+    # via
+    #   unique3d (pyproject.toml)
+    #   peft
+aiofiles==24.1.0
+    # via gradio
+aiohappyeyeballs==2.6.1
+    # via aiohttp
+aiohttp==3.12.15
+    # via fsspec
+aiosignal==1.4.0
+    # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
+antlr4-python3-runtime==4.9.3
+    # via omegaconf
+anyio==4.10.0
+    # via
+    #   gradio
+    #   httpx
+    #   starlette
+async-timeout==5.0.1
+    # via aiohttp
+attrs==25.3.0
+    # via
+    #   aiohttp
+    #   jsonschema
+    #   referencing
+brotli==1.1.0
+    # via gradio
+certifi==2025.8.3
+    # via
+    #   httpcore
+    #   httpx
+    #   requests
+    #   sentry-sdk
+charset-normalizer==3.4.3
+    # via requests
+click==8.2.1
+    # via
+    #   typer
+    #   uvicorn
+    #   wandb
+coloredlogs==15.0.1
+    # via onnxruntime-gpu
+dataclasses-json==0.6.7
+    # via pygltflib
+datasets==4.0.0
+    # via unique3d (pyproject.toml)
+deprecated==1.2.18
+    # via pygltflib
+diffusers==0.33.1
+    # via unique3d (pyproject.toml)
+dill==0.3.8
+    # via
+    #   datasets
+    #   multiprocess
+exceptiongroup==1.3.0
+    # via anyio
+fastapi==0.116.1
+    # via gradio
+ffmpy==0.6.1
+    # via gradio
+filelock==3.19.1
+    # via
+    #   datasets
+    #   diffusers
+    #   huggingface-hub
+    #   torch
+    #   transformers
+    #   triton
+fire==0.7.1
+    # via unique3d (pyproject.toml)
+flatbuffers==25.2.10
+    # via onnxruntime-gpu
+frozenlist==1.7.0
+    # via
+    #   aiohttp
+    #   aiosignal
+fsspec==2025.3.0
+    # via
+    #   datasets
+    #   gradio-client
+    #   huggingface-hub
+    #   torch
+fvcore==0.1.5.post20221221
+    # via pytorch3d
+gitdb==4.0.12
+    # via gitpython
+gitpython==3.1.45
+    # via wandb
+gradio==5.44.1
+    # via
+    #   unique3d (pyproject.toml)
+    #   spaces
+gradio-client==1.12.1
+    # via gradio
+groovy==0.1.2
+    # via gradio
+h11==0.16.0
+    # via
+    #   httpcore
+    #   uvicorn
+hf-transfer==0.1.9
+    # via unique3d (pyproject.toml)
+hf-xet==1.1.9
+    # via huggingface-hub
+httpcore==1.0.9
+    # via httpx
+httpx==0.28.1
+    # via
+    #   gradio
+    #   gradio-client
+    #   safehttpx
+    #   spaces
+huggingface-hub==0.34.4
+    # via
+    #   accelerate
+    #   datasets
+    #   diffusers
+    #   gradio
+    #   gradio-client
+    #   peft
+    #   tokenizers
+    #   transformers
+humanfriendly==10.0
+    # via coloredlogs
+idna==3.10
+    # via
+    #   anyio
+    #   httpx
+    #   requests
+    #   yarl
+imageio==2.37.0
+    # via scikit-image
+importlib-metadata==8.7.0
+    # via diffusers
+iopath==0.1.10
+    # via
+    #   fvcore
+    #   pytorch3d
+jaxtyping==0.3.2
+    # via unique3d (pyproject.toml)
+jinja2==3.1.6
+    # via
+    #   gradio
+    #   torch
+jsonschema==4.25.1
+    # via rembg
+jsonschema-specifications==2025.4.1
+    # via jsonschema
+lazy-loader==0.4
+    # via scikit-image
+llvmlite==0.44.0
+    # via numba
+markdown-it-py==4.0.0
+    # via rich
+markupsafe==3.0.2
+    # via
+    #   gradio
+    #   jinja2
+marshmallow==3.26.1
+    # via dataclasses-json
+mdurl==0.1.2
+    # via markdown-it-py
+mpmath==1.3.0
+    # via sympy
+multidict==6.6.4
+    # via
+    #   aiohttp
+    #   yarl
+multiprocess==0.70.16
+    # via datasets
+mypy-extensions==1.1.0
+    # via typing-inspect
+networkx==3.4.2
+    # via
+    #   scikit-image
+    #   torch
+numba==0.61.2
+    # via pymatting
+numpy==1.26.4
+    # via
+    #   unique3d (pyproject.toml)
+    #   accelerate
+    #   datasets
+    #   diffusers
+    #   fvcore
+    #   gradio
+    #   imageio
+    #   numba
+    #   onnxruntime-gpu
+    #   opencv-python-headless
+    #   pandas
+    #   peft
+    #   pymatting
+    #   pymeshlab
+    #   rembg
+    #   scikit-image
+    #   scipy
+    #   tifffile
+    #   transformers
+    #   trimesh
+    #   xformers
+nvidia-cublas-cu12==12.1.3.1
+    # via
+    #   nvidia-cudnn-cu12
+    #   nvidia-cusolver-cu12
+    #   torch
+nvidia-cuda-cupti-cu12==12.1.105
+    # via torch
+nvidia-cuda-nvrtc-cu12==12.1.105
+    # via torch
+nvidia-cuda-runtime-cu12==12.1.105
+    # via torch
+nvidia-cudnn-cu12==8.9.2.26
+    # via torch
+nvidia-cufft-cu12==11.0.2.54
+    # via torch
+nvidia-curand-cu12==10.3.2.106
+    # via torch
+nvidia-cusolver-cu12==11.4.5.107
+    # via torch
+nvidia-cusparse-cu12==12.1.0.106
+    # via
+    #   nvidia-cusolver-cu12
+    #   torch
+nvidia-nccl-cu12==2.19.3
+    # via torch
+nvidia-nvjitlink-cu12==12.9.86
+    # via
+    #   nvidia-cusolver-cu12
+    #   nvidia-cusparse-cu12
+nvidia-nvtx-cu12==12.1.105
+    # via torch
+omegaconf==2.3.0
+    # via unique3d (pyproject.toml)
+onnxruntime-gpu==1.22.0
+    # via rembg
+opencv-python-headless==4.11.0.86
+    # via
+    #   unique3d (pyproject.toml)
+    #   rembg
+orjson==3.11.3
+    # via gradio
+packaging==25.0
+    # via
+    #   accelerate
+    #   datasets
+    #   gradio
+    #   gradio-client
+    #   huggingface-hub
+    #   lazy-loader
+    #   marshmallow
+    #   onnxruntime-gpu
+    #   peft
+    #   pooch
+    #   scikit-image
+    #   spaces
+    #   transformers
+    #   wandb
+pandas==2.3.2
+    # via
+    #   datasets
+    #   gradio
+peft==0.17.1
+    # via unique3d (pyproject.toml)
+pillow==11.3.0
+    # via
+    #   diffusers
+    #   fvcore
+    #   gradio
+    #   imageio
+    #   pymatting
+    #   rembg
+    #   scikit-image
+platformdirs==4.4.0
+    # via
+    #   pooch
+    #   wandb
+pooch==1.8.2
+    # via rembg
+portalocker==3.2.0
+    # via iopath
+propcache==0.3.2
+    # via
+    #   aiohttp
+    #   yarl
+protobuf==6.32.0
+    # via
+    #   onnxruntime-gpu
+    #   wandb
+psutil==5.9.8
+    # via
+    #   accelerate
+    #   peft
+    #   spaces
+pyarrow==21.0.0
+    # via datasets
+pydantic==2.11.7
+    # via
+    #   fastapi
+    #   gradio
+    #   spaces
+    #   wandb
+pydantic-core==2.33.2
+    # via pydantic
+pydub==0.25.1
+    # via gradio
+pygltflib==1.16.5
+    # via unique3d (pyproject.toml)
+pygments==2.19.2
+    # via rich
+pymatting==1.1.14
+    # via rembg
+pymeshlab==2023.12.post3
+    # via unique3d (pyproject.toml)
+python-dateutil==2.9.0.post0
+    # via pandas
+python-multipart==0.0.20
+    # via gradio
 pytorch3d @ https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu121_pyt221/pytorch3d-0.7.6-cp310-cp310-linux_x86_64.whl
+    # via unique3d (pyproject.toml)
+pytz==2025.2
+    # via pandas
+pyyaml==6.0.2
+    # via
+    #   accelerate
+    #   datasets
+    #   fvcore
+    #   gradio
+    #   huggingface-hub
+    #   omegaconf
+    #   peft
+    #   transformers
+    #   wandb
+    #   yacs
+referencing==0.36.2
+    # via
+    #   jsonschema
+    #   jsonschema-specifications
+regex==2025.8.29
+    # via
+    #   diffusers
+    #   transformers
+rembg==2.0.62
+    # via unique3d (pyproject.toml)
+requests==2.32.5
+    # via
+    #   datasets
+    #   diffusers
+    #   huggingface-hub
+    #   pooch
+    #   spaces
+    #   transformers
+    #   wandb
+rich==14.1.0
+    # via typer
+rpds-py==0.27.1
+    # via
+    #   jsonschema
+    #   referencing
+ruff==0.12.11
+    # via gradio
+safehttpx==0.1.6
+    # via gradio
+safetensors==0.6.2
+    # via
+    #   accelerate
+    #   diffusers
+    #   peft
+    #   transformers
+scikit-image==0.25.2
+    # via rembg
+scipy==1.15.3
+    # via
+    #   pymatting
+    #   rembg
+    #   scikit-image
+semantic-version==2.10.0
+    # via gradio
+sentry-sdk==2.35.1
+    # via wandb
+shellingham==1.5.4
+    # via typer
+six==1.17.0
+    # via python-dateutil
+smmap==5.0.2
+    # via gitdb
+sniffio==1.3.1
+    # via anyio
+spaces==0.40.1
+    # via unique3d (pyproject.toml)
+starlette==0.47.3
+    # via
+    #   fastapi
+    #   gradio
+sympy==1.14.0
+    # via
+    #   onnxruntime-gpu
+    #   torch
+tabulate==0.9.0
+    # via fvcore
+termcolor==3.1.0
+    # via
+    #   fire
+    #   fvcore
+tifffile==2025.5.10
+    # via scikit-image
+tokenizers==0.21.4
+    # via transformers
+tomlkit==0.13.3
+    # via gradio
 torch==2.2.0
-accelerate
-datasets
-diffusers>=0.26.3
-fire
-gradio
-jaxtyping
-numba
-numpy<2
-omegaconf>=2.3.0
-opencv_python
-opencv_python_headless
-peft
-Pillow
-pygltflib
-pymeshlab>=2023.12
-rembg[gpu]==2.0.62
-torch_scatter @ https://data.pyg.org/whl/torch-2.2.0%2Bcu121/torch_scatter-2.1.2%2Bpt22cu121-cp310-cp310-linux_x86_64.whl
-tqdm
+    # via
+    #   unique3d (pyproject.toml)
+    #   accelerate
+    #   peft
+    #   xformers
+torch-scatter @ https://data.pyg.org/whl/torch-2.2.0%2Bcu121/torch_scatter-2.1.2%2Bpt22cu121-cp310-cp310-linux_x86_64.whl
+    # via unique3d (pyproject.toml)
+tqdm==4.67.1
+    # via
+    #   datasets
+    #   fvcore
+    #   huggingface-hub
+    #   iopath
+    #   peft
+    #   rembg
+    #   transformers
 transformers==4.48.2
-trimesh
-typeguard
-wandb
-xformers
-ninja
+    # via
+    #   unique3d (pyproject.toml)
+    #   peft
+trimesh==4.7.4
+    # via unique3d (pyproject.toml)
+triton==2.2.0
+    # via torch
+typeguard==4.4.4
+    # via unique3d (pyproject.toml)
+typer==0.17.3
+    # via gradio
+typing-extensions==4.15.0
+    # via
+    #   aiosignal
+    #   anyio
+    #   exceptiongroup
+    #   fastapi
+    #   gradio
+    #   gradio-client
+    #   huggingface-hub
+    #   iopath
+    #   multidict
+    #   pydantic
+    #   pydantic-core
+    #   referencing
+    #   spaces
+    #   starlette
+    #   torch
+    #   typeguard
+    #   typer
+    #   typing-inspect
+    #   typing-inspection
+    #   uvicorn
+    #   wandb
+typing-inspect==0.9.0
+    # via dataclasses-json
+typing-inspection==0.4.1
+    # via pydantic
+tzdata==2025.2
+    # via pandas
+urllib3==2.5.0
+    # via
+    #   requests
+    #   sentry-sdk
+uvicorn==0.35.0
+    # via gradio
+wadler-lindig==0.1.7
+    # via jaxtyping
+wandb==0.21.3
+    # via unique3d (pyproject.toml)
+websockets==15.0.1
+    # via gradio-client
+wrapt==1.17.3
+    # via deprecated
+xformers==0.0.24
+    # via unique3d (pyproject.toml)
+xxhash==3.5.0
+    # via datasets
+yacs==0.1.8
+    # via fvcore
+yarl==1.20.1
+    # via aiohttp
+zipp==3.23.0
+    # via importlib-metadata