torch<2.15,>=2.13
numpy
packaging
pillow>=9.4
safetensors
tqdm
kohaku-engine
anyschedule
kohakuvault
lycoris-lora>=4.1.2
diffusers>=0.38
transformers>=5.3
accelerate

[baselines]
flash-linear-attention

[baselines:sys_platform != "win32"]
causal-conv1d

[bench]
matplotlib
pandas

[board]
kohakuboard

[cuda]
kohakumoss[kernels-cuda,lightning,quant]

[dev]
pytest
ruff
black
lpips

[kernels]
kohakumoss[kernels-cuda]

[kernels-cuda]
tilelang>=0.1.15

[kernels-cuda:sys_platform == "win32"]
triton-windows<3.9,>=3.7

[kernels-xpu]

[lightning]
lightning>=2.5

[optim]
prodigyopt
schedulefree

[parquet]
pyarrow

[quant]
bitsandbytes

[wandb]
wandb

[xpu]
kohakumoss[kernels-xpu,lightning,quant]
