Skip to content

GitLab

Unverified Commit 0d9917f7 authored by abhishek thakur's avatar abhishek thakur Committed by GitHub
Browse files

Update peft + transformers + accelerate + bnb + safetensors (#1646)

parent 8a5bcba2
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
server/pyproject.toml
View file @ 0d9917f7
......@@ -15,9 +15,9 @@ grpcio-status = "^1.51.1"
grpcio-reflection = "^1.51.1"
grpc-interceptor = "^0.15.0"
typer = "^0.6.1"
accelerate = { version = "^0.25.0", optional = true }
bitsandbytes = { version = "^0.41.1", optional = true }
safetensors = "^0.3.2"
accelerate = { version = "^0.28.0", optional = true }
bitsandbytes = { version = "^0.43.0", optional = true }
safetensors = "^0.4.1"
loguru = "^0.6.0"
opentelemetry-api = "^1.15.0"
opentelemetry-exporter-otlp = "^1.15.0"
......@@ -26,11 +26,11 @@ hf-transfer = "^0.1.2"
sentencepiece = "^0.1.97"
tokenizers = "^0.15.0"
huggingface-hub = "^0.19.3"
transformers = "^4.37.1"
transformers = "^4.38.2"
einops = "^0.6.1"
texttable = { version = "^1.6.7", optional = true }
datasets = { version = "^2.14.0", optional = true }
peft = { version = "^0.8.2", optional = true }
peft = { version = "^0.9.0", optional = true }
torch = { version = "^2.1.1", optional = true }
scipy = "^1.11.1"
pillow = "^10.0.0"
......
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