PyPI page
Home page
Author:
hiyouga
License:
Apache 2.0 License
Summary:
Easy-to-use LLM fine-tuning framework
Latest version:
0.7.1
Required dependencies:
accelerate
|
datasets
|
einops
|
fastapi
|
fire
|
gradio
|
matplotlib
|
packaging
|
peft
|
protobuf
|
pydantic
|
pyyaml
|
scipy
|
sentencepiece
|
sse-starlette
|
transformers
|
trl
|
uvicorn
Optional dependencies:
aqlm
|
auto-gptq
|
autoawq
|
badam
|
bitsandbytes
|
deepspeed
|
galore-torch
|
jieba
|
modelscope
|
nltk
|
optimum
|
rouge-chinese
|
ruff
|
tiktoken
|
torch
|
transformers-stream-generator
|
vllm
Downloads last day:
84
Downloads last week:
236
Downloads last month:
617