Skip to content

GitLab

Commit 94ce276c authored by Zach Nussbaum's avatar Zach Nussbaum
Browse files

fix: reqs for peft + sentencepiece tokenizers (llama model)

parent 2b9a6e94
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
setup.py
View file @ 94ce276c
...@@ -26,6 +26,7 @@ setuptools.setup( ...@@ -26,6 +26,7 @@ setuptools.setup(
"numexpr", "numexpr",
"openai>=0.6.4", "openai>=0.6.4",
"omegaconf>=2.2", "omegaconf>=2.2",
"peft>=0.2.0"
"pybind11>=2.6.2", "pybind11>=2.6.2",
"pycountry", "pycountry",
"pytablewriter", "pytablewriter",
...@@ -41,5 +42,6 @@ setuptools.setup( ...@@ -41,5 +42,6 @@ setuptools.setup(
extras_require={ extras_require={
"dev": ["black", "flake8", "pre-commit", "pytest", "pytest-cov"], "dev": ["black", "flake8", "pre-commit", "pytest", "pytest-cov"],
"multilingual": ["nagisa>=0.2.7", "jieba>=0.42.1"], "multilingual": ["nagisa>=0.2.7", "jieba>=0.42.1"],
"sentencepiece": ["sentencepiece>=0.1.98", "protobuf>=4.22.1"]
}, },
) )
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment