Skip to content

Commit

Permalink
change dependencies so that we can upload on pypi (#22)
Browse files Browse the repository at this point in the history
  • Loading branch information
NathanHB authored Feb 7, 2024
1 parent 59891e0 commit f8bd2ab
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 f8bd2ab

Please sign in to comment.