* diffusion training plugin * cleanup * nits * fixes + improvements * add back in reinit_weights (clobbered?); masking / pretrain fixes * nits * cleanup; tests draft * sample generation, tests fixes * fixes * nits * add inference support; add auto-mask token support * nits * nits * progress * simplify logging * lint * prefix args with diffusion_ * coderabbito * tests fix * nit * nits * cleanup + nits * nits * fix SFT sample gen * fixes * fix * comments * comments * lint * reward model lora fix * cleanup; fix pretraining_dataset case * gradio inference * update cfgs * update cfgs * train, generation parity, cleanup * fix * simplify * test * test fix
36 lines
813 B
YAML
36 lines
813 B
YAML
default_language_version:
|
|
python: python3
|
|
|
|
repos:
|
|
- repo: https://github.com/pre-commit/pre-commit-hooks
|
|
rev: v6.0.0
|
|
hooks:
|
|
- id: check-yaml
|
|
- id: end-of-file-fixer
|
|
- id: trailing-whitespace
|
|
- id: no-commit-to-branch
|
|
args: ['--branch', 'main']
|
|
- repo: https://github.com/astral-sh/ruff-pre-commit
|
|
rev: v0.12.12
|
|
hooks:
|
|
- id: ruff
|
|
args: [--fix, --select, I]
|
|
- id: ruff-format
|
|
- repo: https://github.com/pre-commit/mirrors-mypy
|
|
rev: v1.17.1
|
|
hooks:
|
|
- id: mypy
|
|
additional_dependencies:
|
|
[
|
|
'types-PyYAML',
|
|
'pydantic>=2.5.3',
|
|
]
|
|
- repo: https://github.com/PyCQA/bandit
|
|
rev: 1.8.6
|
|
hooks:
|
|
- id: bandit
|
|
args: [
|
|
'--ini',
|
|
'.bandit',
|
|
]
|