Skip to content

Instantly share code, notes, and snippets.

@poptree
Created October 5, 2024 05:48
Show Gist options
  • Save poptree/ff8935058323b5158311ad5e00ac4611 to your computer and use it in GitHub Desktop.
Save poptree/ff8935058323b5158311ad5e00ac4611 to your computer and use it in GitHub Desktop.
[build-system]
requires = ["setuptools >= 61.0"]
build-backend = "setuptools.build_meta"
[project]
version = "0.0.1-dev"
name = "simple_gaussian"
requires-python = ">=3.10"
dependencies = [
"lightning==2.0.0",
"omegaconf==2.3.0",
"jaxtyping>=0.2.20",
"typeguard==2.13.3",
#"nerfacc@git+https://github.com/KAIR-BAIR/nerfacc#egg=v0.5.2",
#"tinycudann@git+https://github.com/NVlabs/tiny-cuda-nn/#subdirectory=bindings/torch",
"diffusers",
"peft",
"transformers>=4.35.0",
"accelerate",
"opencv-python",
"tensorboard",
"matplotlib",
"imageio>=2.28.0",
"imageio[ffmpeg]",
"Pillow==10.1.0",
"torchmcubes@git+https://github.com/tatsy/torchmcubes",
"rembg[gpu]",
"nvdiffrast@git+https://github.com/NVlabs/nvdiffrast",
"libigl",
"xatlas",
"trimesh[easy]",
"networkx",
"pysdf",
"PyMCubes",
"wandb",
"gradio",
"envlight@git+https://github.com/ashawkey/envlight",
"torchmetrics",
"xformers",
"bitsandbytes==0.38.1",
"sentencepiece",
"safetensors",
"huggingface_hub>=0.20.0", # fixed insecure_hashing error
"einops",
"kornia",
"taming-transformers-rom1504",
"clip@git+https://github.com/openai/CLIP",
"controlnet_aux",
"torch>=1.13.1,<2.2",
"torchvision>=0.14.1",
"torchmetrics[image]>=1.0.1",
"typing_extensions>=4.4.0",
"pytorch-msssim",
"easydict",
"datasets",
"torch-scatter",
# ScanNet depth dataset
"OpenEXR",
# pytorch3d
"fvcore",
# Updating the link if the version of pytorch is different.
"pytorch3d@https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py310_cu118_pyt201/pytorch3d-0.7.4-cp310-cp310-linux_x86_64.whl",
]
[project.optional-dependencies]
dev = [
"black",
"mypy",
"pylint",
"pre-commit",
"pytest",
"pytest-xdist"
]
[tool.setuptools.packages.find]
include = ["spgs"]
[tool.pytest.ini_options]
# addopts = "-n=4 --typeguard-packages=inv3d --disable-warnings"
addopts = "-n=4 --disable-warnings" # Typeguard has bug, it cannot be used to test the tensor on the cuda.
testpaths = [
"tests",
]
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment