| [build-system] | |
| requires = ["setuptools>=61.0"] | |
| build-backend = "setuptools.backends._legacy:_Backend" | |
| [project] | |
| name = "cdd" | |
| version = "1.0.0" | |
| description = "Constrained Discrete Diffusion (CDD) - Constrained Language Generation with Discrete Diffusion Models" | |
| readme = "README.md" | |
| license = {text = "MIT"} | |
| requires-python = ">=3.9" | |
| dependencies = [ | |
| "torch>=2.0.0", | |
| "transformers>=4.35.0", | |
| "datasets>=2.14.0", | |
| "einops>=0.7.0", | |
| "tqdm>=4.65.0", | |
| "numpy>=1.24.0", | |
| ] | |
| [project.optional-dependencies] | |
| flash-attn = ["flash-attn>=2.5.0"] | |
| rdkit = ["rdkit-pypi>=2022.9.0"] | |
| eval = ["nltk>=3.8.0"] | |
| all = ["flash-attn>=2.5.0", "rdkit-pypi>=2022.9.0", "nltk>=3.8.0"] | |
| [tool.setuptools.packages.find] | |
| include = ["cdd*"] | |