| [build-system] | |
| requires = ["setuptools>=68.0", "wheel"] | |
| build-backend = "setuptools.build_meta" | |
| [project] | |
| name = "llada-guard" | |
| version = "0.1.0" | |
| description = "Safety content classifier based on LLaDA diffusion language model" | |
| requires-python = ">=3.10" | |
| license = { text = "MIT" } | |
| dependencies = [ | |
| "torch>=2.0", | |
| "transformers>=4.42,<5", | |
| "peft>=0.10", | |
| "accelerate", | |
| "tqdm", | |
| ] | |
| [tool.setuptools.packages.find] | |
| include = ["llada_guard*", "llada_guard_infer*"] | |