[tool.poetry]
name = "space-llama-13b-japanese-lora-v0-1ep"
version = "0.1.0"
description = ""
authors = ["Masanori HIRANO <masa.hirano.1996@gmail.com>"]
license = "other"
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.9"
peft = "^0.3.0"
gradio = "^3.23.0"
torch = "^2.0.1"
huggingface-hub = "^0.14.1"
sentencepiece = "^0.1.99"
bitsandbytes = "^0.38.1"
accelerate = "^0.19.0"
fschat = "^0.2.3"
transformers = "^4.29.2"


[tool.poetry.group.dev.dependencies]
black = "^23.3.0"
isort = "^5.12.0"
mypy = "^1.3.0"
flake8 = "^6.0.0"
pyproject-flake8 = "^6.0.0.post1"

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

[tool.isort]
profile = 'black'
force_single_line = true
skip = [
    ".git",
    "__pycache__",
    "docs",
    "build",
    "dist",
    "examples",
    ".venv",
    "tests/examples"
]

[tool.mypy]
disallow_untyped_defs = true
ignore_missing_imports = true

[tool.flake8]
ignore = "E203,E231,E501,W503"
max-line-length = 88
exclude = [
    ".git",
    "__pycache__",
    "docs",
    "build",
    "dist",
    "examples",
    ".venv",
    "__init__.py"
    ]
select = "B,B950,C,E,F,W"