mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2025-09-10 09:04:36 +00:00
fix for gemma3n
This commit is contained in:
parent
b8c1fc7c9e
commit
c45b8dc56f
4 changed files with 58 additions and 35 deletions
|
@ -2287,6 +2287,11 @@ ModelLoadResult gpttype_load_model(const load_model_inputs inputs, FileFormat in
|
|||
}
|
||||
//handle override tensor
|
||||
std::string tensoroverrides = inputs.override_tensors;
|
||||
if(file_format_meta.model_architecture==GGUFArch::ARCH_GEMMA3N)
|
||||
{
|
||||
std::string forced = "per_layer_token_embd.weight=CPU"; //this tensor on gpu is problematic on unsloth q4_0
|
||||
tensoroverrides = (tensoroverrides=="" ? forced: (forced+","+tensoroverrides));
|
||||
}
|
||||
if(tensoroverrides!="" && ggml_backend_dev_count()>1)
|
||||
{
|
||||
printf("Handling Override Tensors for backends: ");
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue