mirror of
https://github.com/LostRuins/koboldcpp.git
synced 2026-09-04 03:51:17 +02:00
only accept qwen image pruned models matching 40 or 41 layers
This commit is contained in:
@@ -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);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user