[Backend] Add Llamacpp backend #2975
angt
force pushed
from
78857ecb
to
fb744095
1 year ago
mfuntowicz
changed the title Add Llamacpp backend [Backend] Add Llamacpp backend 1 year ago
angt
force pushed
from
76516da7
to
58775423
1 year ago
Add llamacpp backend
95e221ee
Get rid of llama_batch_get_one()
bd0cc990
Use max_batch_total_tokens
3eb4823f
Handle max_batch_size
e7facf69
Add some input validation checks
a7b4b04c
Handle ctx args & fix sampling
8d2dfdf6
Add GPU args
f3887479
Add --defrag-threshold
e07835c5
Add a stupid batch mechanism
d6ded897
Cleanup
390f0ec0
Add --numa
7a3ed417
Fix args
3f199134
Enable flash attention by default
ae5bb789
Add --offload-kqv
e88a527f
Fix batch_pos
f38c34ae
backend(llama): add CUDA Dockerfile_llamacpp for now
960c12bd
Only export the latest logits
161280f3
Output real logprobs
2a51e415
Fix batching
96434a1e
Fix seq iterations
27534d8e
Auto-detect n_threads when not provided
c8505fb3
Clear request cache after completion
8ed362d0
Remove warmup
104a968d
Cleanup
ea28332b
backend(llama): add CUDA architectures build argument for Dockerfile
e6a8d339
Add specific args for batch
bfb8e03e
Add --type-v & --type-k
38b33e96
Bump llamacpp to b4623
207041a9
Disable graceful shutdown in debug mode
d883109d
Update Dockerfile_llamacpp
df2a4fbb
angt
force pushed
from
58775423
to
df2a4fbb
1 year ago
Cleanup Dockerfile
906c265a
Update Cargo.lock
e0075295
Update args
d3a772a8
Simplify batching logic
dbee8041
Set TGI_LLAMA_PKG_CUDA from CUDA_VERSION
c52f0835
Rename bindings
051ff2d5
Remove n_ctx
09a745f1
Make max_batch_total_tokens optional
5b777877
Ensure all samplers are freed on error
695b1292
Initialize penalty_last_n with llamacpp default value
0f62401b
Cleanup
f22e2fb5
Improve default settings
b3e40c4b
Add doc
1641c22a
Update docs
e4d5fa7e
Thanks clippy
fb81c0d1
Thanks cargo fmt
2b0d99c1
Update docs
8bc10d37
angt
marked this pull request as ready for review 1 year ago
Do not use HOSTNAME env
7bff88bb
Bump llama.cpp & cuda
df723e64
Fix requirements.txt
5367d94f
Fix fmt
809e288b
Enable KQV offload by default
3b1b049b
Remove Ngrok tunneling
acca9c3e
angt
dismissed their stale review
via acca9c3e
1 year ago
angt
force pushed
from
bc935dc3
to
acca9c3e
1 year ago
Remove .cargo/config.toml
0d27ee74
Fix Dockerfile
4841f71a
Add missing cuda prefix
b6cfa0fb
Handle custom llama.cpp dir
6bdb644f
Cleanup
0702e0bf
Add README.md
508d47f8
Add HF transfer
14014182
Fix bool args
b77d05d3
Update doc
d96a7770
Update doc
5fb4afbf
Hugoch
approved these changes
on 2025-02-14
angt
merged
cfd4fbb4
into main 1 year ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub