From f1382d34591b273b7f080ea05ee03db9177afcd2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=8A=B1=E4=B8=80?= Date: Wed, 24 Sep 2025 10:38:09 +0800 Subject: [PATCH] fix: mid aux hidden layer id calculattion in online mode --- specforge/core/eagle3.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/specforge/core/eagle3.py b/specforge/core/eagle3.py index 2886aec5..8a8e5702 100644 --- a/specforge/core/eagle3.py +++ b/specforge/core/eagle3.py @@ -120,7 +120,7 @@ def _prepare_data( aux_layer_ids = eagle_config["eagle_aux_hidden_state_layer_ids"] assert len(aux_layer_ids) == 3, "EAGLE3 requires 3 aux layers" else: - aux_layer_ids = [1, num_layers // 2, num_layers - 4] + aux_layer_ids = [1, num_layers // 2 - 1, num_layers - 4] low_aux_layer = aux_layer_ids[0] + offset mid_aux_layer = aux_layer_ids[1] + offset