1127 update to latest
This commit is contained in:
51
dllm/pyproject.toml
Normal file
51
dllm/pyproject.toml
Normal file
@ -0,0 +1,51 @@
|
||||
[build-system]
|
||||
requires = ["setuptools>=61", "wheel"]
|
||||
build-backend = "setuptools.build_meta"
|
||||
|
||||
[tool.setuptools]
|
||||
packages = ["dllm"]
|
||||
|
||||
[project]
|
||||
name = "dllm"
|
||||
version = "0.1.0"
|
||||
description = "dLLM: Simple Diffusion Language Modeling"
|
||||
readme = "README.md"
|
||||
requires-python = ">=3.10"
|
||||
dependencies = [
|
||||
"transformers==4.57.0",
|
||||
"accelerate==1.11.0",
|
||||
"deepspeed==0.18.0",
|
||||
"peft==0.17.1",
|
||||
"datasets==4.2.0",
|
||||
"sentencepiece==0.2.0",
|
||||
"tyro",
|
||||
"wandb",
|
||||
"omegaconf",
|
||||
"tqdm",
|
||||
"matplotlib",
|
||||
"pytest",
|
||||
"rich",
|
||||
]
|
||||
|
||||
[project.optional-dependencies]
|
||||
optional = [
|
||||
"bitsandbytes==0.48.1",
|
||||
"vllm==0.8.5.post1",
|
||||
"flash-attn==2.8.3",
|
||||
]
|
||||
|
||||
[tool.black]
|
||||
line-length = 88
|
||||
exclude = '''
|
||||
(
|
||||
(^|/)dllm/pipelines/llada/models(/|$)|
|
||||
(^|/)dllm/pipelines/dream/models(/|$)|
|
||||
(^|/)dllm/pipelines/rnd/models(/|$)|
|
||||
(^|/)lm-evaluation-harness(/|$)
|
||||
)
|
||||
'''
|
||||
|
||||
[tool.pytest.ini_options]
|
||||
testpaths = ["scripts/tests"]
|
||||
python_files = ["test_*.py"]
|
||||
addopts = "-v -ra"
|
||||
Reference in New Issue
Block a user