ExternalVendorCode/LocalAI
1
0
Fork 0
mirror of https://github.com/mudler/LocalAI.git synced 2025-05-02 08:43:10 +00:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

feat(llama.cpp): add flash_attention and no_kv_offloading (#2310)

Browse Source 
feat(llama.cpp): add flash_attn and no_kv_offload

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
Ettore Di Giacinto 2024-05-13 19:07:51 +02:00 committed by GitHub
parent 7123d07456
commit e49ea0123b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/backend.proto
View File

@ -212,6 +212,9 @@ message ModelOptions {
float YarnBetaSlow = 47; float YarnBetaSlow = 47;
string Type = 49; string Type = 49;
bool FlashAttention = 56;
bool NoKVOffload = 57;
} }
message Result { message Result {

3
backend/cpp/llama/grpc-server.cpp
View File

@ -2254,6 +2254,9 @@ static void params_parse(const backend::ModelOptions* request,
} }
params.use_mlock = request->mlock(); params.use_mlock = request->mlock();
params.use_mmap = request->mmap(); params.use_mmap = request->mmap();
params.flash_attn = request->flashattention();
params.no_kv_offload = request->nokvoffload();
params.embedding = request->embeddings(); params.embedding = request->embeddings();
if (request->ropescaling() == "none") { params.rope_scaling_type = LLAMA_ROPE_SCALING_TYPE_NONE; } if (request->ropescaling() == "none") { params.rope_scaling_type = LLAMA_ROPE_SCALING_TYPE_NONE; }

2
core/backend/options.go
View File

@ -77,6 +77,8 @@ func gRPCModelOpts(c config.BackendConfig) *pb.ModelOptions {
MaxModelLen: int32(c.MaxModelLen), MaxModelLen: int32(c.MaxModelLen),
TensorParallelSize: int32(c.TensorParallelSize), TensorParallelSize: int32(c.TensorParallelSize),
MMProj: c.MMProj, MMProj: c.MMProj,
FlashAttention: c.FlashAttention,
NoKVOffload: c.NoKVOffloading,
YarnExtFactor: c.YarnExtFactor, YarnExtFactor: c.YarnExtFactor,
YarnAttnFactor: c.YarnAttnFactor, YarnAttnFactor: c.YarnAttnFactor,
YarnBetaFast: c.YarnBetaFast, YarnBetaFast: c.YarnBetaFast,

3
core/config/backend_config.go
View File

@ -132,6 +132,9 @@ type LLMConfig struct {
TensorParallelSize int `yaml:"tensor_parallel_size"` // vLLM TensorParallelSize int `yaml:"tensor_parallel_size"` // vLLM
MMProj string `yaml:"mmproj"` MMProj string `yaml:"mmproj"`
FlashAttention bool `yaml:"flash_attention"`
NoKVOffloading bool `yaml:"no_kv_offloading"`
RopeScaling string `yaml:"rope_scaling"` RopeScaling string `yaml:"rope_scaling"`
ModelType string `yaml:"type"` ModelType string `yaml:"type"`