pyproject.toml

[tool.poetry]
name = "lora-instruct"
version = "0.1.0"
description = ""
authors = ["Han Lee <lee.hanchung@gmail.com>"]
license = "LICENSE"
readme = "README.md"
packages = [{include = "lora_instruct"}]

[tool.poetry.dependencies]
python = "^3.10"
sentencepiece = "^0.1.99"
accelerate = "^0.25.0"
loralib = "^0.1.1"
bitsandbytes = "^0.41.3"
datasets = "^2.15.0"
fire = "^0.5.0"
peft = "^0.7.0"
transformers = "^4.35.2"
einops = "^0.7.0"
python-dotenv = "^1.0.0"
wandb = "^0.16.1"
vllm = "^0.2.3"
openai = "^1.3.8"
tiktoken = "^0.5.2"
num2words = "^0.5.13"

[tool.poetry.group.dev.dependencies]
ipykernel = "^6.22.0"
ipywidgets = "^8.0.6"
pytest = "^7.3.1"
ruff = "^0.0.292"

[tool.ruff]
select = [
    "E",  # pycodestyle errors
    "W",  # pycodestyle warnings
    "F",  # pyflakes
    "I",  # isort
    "C",  # flake8-comprehensions
    "B",  # flake8-bugbear
]
ignore = [
    "E501",  # line too long, handled by black
    "B008",  # do not perform function calls in argument defaults
    "C901",  # too complex
]

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"