mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2026-05-07 17:22:04 +00:00
only accept qwen image pruned models matching 40 or 41 layers
This commit is contained in:
parent
bc762fe9b4
commit
bc09f34f66
1 changed files with 3 additions and 3 deletions
|
|
@ -508,11 +508,11 @@ namespace Qwen {
|
|||
}
|
||||
}
|
||||
}
|
||||
if (num_layers < model_layers) {
|
||||
LOG_INFO("Qwen Image: some layers missing, assuming pruned model");
|
||||
if (num_layers == 40 || num_layers == 41) {
|
||||
qwen_image_params.num_layers = num_layers;
|
||||
LOG_INFO("Qwen Image: Found %d layers instead of %d, assuming pruned model.",num_layers,model_layers);
|
||||
}
|
||||
|
||||
qwen_image_params.num_layers = num_layers;
|
||||
qwen_image = QwenImageModel(qwen_image_params);
|
||||
qwen_image.init(params_ctx, tensor_types, prefix);
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue