From 3d3f02ef4ae197af7e4a811ecd4ee061c0ab31a0 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Sun, 8 Feb 2026 12:58:05 +0800
Subject: [PATCH] revert layers if fail

---
 gpttype_adapter.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/gpttype_adapter.cpp b/gpttype_adapter.cpp
index 99966924c..059497fa3 100644
--- a/gpttype_adapter.cpp
+++ b/gpttype_adapter.cpp
@@ -2582,6 +2582,11 @@ ModelLoadResult gpttype_load_model(const load_model_inputs inputs, FileFormat in
             }
             printf("Autofit Success: %d, Autofit Result: ",success);
             print_fitted_params(model_params,llama_ctx_params);
+            if(!success)
+            {
+                //revert to previous
+                model_params.n_gpu_layers = inputs.gpulayers;
+            }
         }
 
         llama_model * llamamodel = llama_model_load_from_file(kcpp_data->model_filename.c_str(), model_params);