apunkt/doc-to-lora
1
0
Fork 0
mirror of https://github.com/SakanaAI/doc-to-lora.git synced 2026-07-23 17:01:04 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

install

Browse source
This commit is contained in:
51616 2025-01-20 11:38:10 +00:00
parent 6e7bc84966
commit 773094cb22
2 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
install.sh
View file

@ -1 +1,2 @@
pip install -e .
pip install -e .
pip install flash-attn --no-build-isolation

13
pyproject.toml
View file

@ -6,9 +6,10 @@ description = ""
readme = "README.md"
requires-python = ">= 3.10"
dependencies = [
"transformers",
"transformers==4.46.3",
"deepspeed==0.15.4",
"accelerate==1.2.1",
"peft",
"accelerate",
"datasets",
"bitsandbytes",
"jupyter",
@ -17,10 +18,16 @@ dependencies = [
"torchmetrics",
"inflect",
"pre-commit",
"trl",
"tensorboardX",
"wandb",
"fasttext-wheel",
"einops",
"jaxtyping",
"liger-kernel",
"tensorboard",
"flask",
"pandas",
"plotly",
]
[build-system]