openai>=1.0.0
pyyaml>=6.0

[all]
squeez[train]
squeez[encoder]

[dev]
squeez[all]
pytest>=7.0.0
ruff>=0.0.270

[docs]
mkdocs-material>=9.0
mkdocstrings[python]>=0.24

[encoder]
torch>=2.1.0
transformers>=5.0.0
datasets>=2.14.0

[local]
torch>=2.1.0
transformers>=5.0.0
peft>=0.7.0

[train]
squeez[local]
datasets>=2.14.0
trl>=0.24.0
unsloth
