llama.cpp/common
Georgi Gerganov f4d2b8846a
llama : add reranking support (#9510)
* py : add XLMRobertaForSequenceClassification [no ci]

* py : fix scalar-tensor conversion [no ci]

* py : fix position embeddings chop [no ci]

* llama : read new cls tensors [no ci]

* llama : add classigication head (wip) [no ci]

* llama : add "rank" pooling type

ggml-ci

* server : add rerank endpoint

ggml-ci

* llama : aboud ggml_repeat during classification

* rerank : cleanup + comments

* server : accept /rerank endpoint in addition to /v1/rerank [no ci]

* embedding : parse special tokens

* jina : support v1 reranker

* vocab : minor style

ggml-ci

* server : initiate tests for later

ggml-ci

* server : add docs

* llama : add comment [no ci]

* llama : fix uninitialized tensors

* ci : add rerank tests

ggml-ci

* add reranking test

* change test data

* Update examples/server/server.cpp

Co-authored-by: Xuan Son Nguyen <thichthat@gmail.com>

* add `--reranking` argument

* update server docs

* llama : fix comment [no ci]

ggml-ci

---------

Co-authored-by: Xuan Son Nguyen <son@huggingface.co>
Co-authored-by: Xuan Son Nguyen <thichthat@gmail.com>
2024-09-28 17:42:03 +03:00
..
cmake llama : reorganize source code + improve CMake (#8006) 2024-06-26 18:33:02 +03:00
arg.cpp llama : add reranking support (#9510) 2024-09-28 17:42:03 +03:00
arg.h common : move arg parser code to arg.cpp (#9388) 2024-09-09 23:36:09 +02:00
base64.hpp llava : expose as a shared library for downstream projects (#3613) 2023-11-07 00:36:23 +03:00
build-info.cpp.in build : link against build info instead of compiling against it (#3879) 2023-11-02 08:50:16 +02:00
CMakeLists.txt common : reimplement logging (#9418) 2024-09-15 20:46:12 +03:00
common.cpp llama : add reranking support (#9510) 2024-09-28 17:42:03 +03:00
common.h llama : add reranking support (#9510) 2024-09-28 17:42:03 +03:00
console.cpp check C++ code with -Wmissing-declarations (#3184) 2023-09-15 15:38:27 -04:00
console.h gguf : new file format with flexible meta data (beta) (#2398) 2023-08-21 23:07:43 +03:00
json-schema-to-grammar.cpp json: restore default additionalProperties to false, fix some pattern escapes (#8180) 2024-06-28 09:26:45 +01:00
json-schema-to-grammar.h JSON: [key] -> .at(key), assert() -> GGML_ASSERT (#7143) 2024-05-08 21:53:08 +02:00
json.hpp json-schema-to-grammar improvements (+ added to server) (#5978) 2024-03-21 11:50:43 +00:00
log.cpp log : add CONT level for continuing previous log entry (#9610) 2024-09-24 10:15:35 +03:00
log.h log : add CONT level for continuing previous log entry (#9610) 2024-09-24 10:15:35 +03:00
ngram-cache.cpp common : reimplement logging (#9418) 2024-09-15 20:46:12 +03:00
ngram-cache.h lookup: fibonacci hashing, fix crashes (#8548) 2024-07-17 23:35:44 +02:00
sampling.cpp sampling : avoid expensive softmax during greedy sampling (#9605) 2024-09-24 09:03:17 +03:00
sampling.h llama : move random seed generation to the samplers (#9398) 2024-09-10 18:04:25 +02:00
stb_image.h common : Update stb_image.h to latest version (#9161) 2024-08-27 08:58:50 +03:00
train.cpp common : reimplement logging (#9418) 2024-09-15 20:46:12 +03:00
train.h sync : ggml (backend v2) (#3912) 2023-11-13 14:16:23 +02:00