[metadata]
name = axolotl
version = 0.1.0
description = You know you're going to axolotl questions
author = Wing Lian
author_email = wing.lian@gmail.com
license = MIT

[options]
package_dir =
    =src
packages = find:
install_requires =
    transformers @ git+https://github.com/huggingface/transformers.git@main
    peft @ git+https://github.com/huggingface/peft.git@main
    attrdict
    fire
    PyYAML == 6.0
    black
    bitsandbytes
    datasets
    accelerate
    sentencepiece
    wandb
    flash-attn
    einops

[options.packages.find]
where = src

[options.extras_require]
gptq_cuda = alpaca_lora_4bit[cuda] @ git+https://github.com/winglian/alpaca_lora_4bit.git@setup_pip#egg=alpaca_lora_4bit[cuda]
gptq_triton = alpaca_lora_4bit[triton] @ git+https://github.com/winglian/alpaca_lora_4bit.git@setup_pip#egg=alpaca_lora_4bit[triton]