File size: 730 Bytes
2d0a056 | 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 | [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*"]
|