Skip to content

Commit

Permalink
Merge branch 'main' into custom-tasks
Browse files Browse the repository at this point in the history
  • Loading branch information
clefourrier committed Feb 8, 2024
2 parents 1b1aa8e + f8bd2ab commit 28170b3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -75,14 +75,14 @@ dependencies = [
]

[project.optional-dependencies]
accelerate = ["accelerate@git+https://github.com/huggingface/accelerate@739b135f8367becb67ffaada12fe76e3aa60fefd"]
accelerate = ["accelerate"]
tgi = ["text-generation==0.6.0"]
optimum = ["optimum==1.12.0"]
# Quantization and adapter weights
quantization = ["bitsandbytes>=0.41.0", "auto-gptq>=0.4.2"]
adapters = ["peft==0.3.0"]
nanotron = [
"nanotron@git+https://github.com/huggingface/nanotron",
"nanotron",
"tensorboardX"
]

Expand Down

0 comments on commit 28170b3

Please sign in to comment.