fixup logging layer

2025-01-15 21:36:14 -05:00
parent 145664d82c
commit 79ae776102
1 changed files with 1 additions and 1 deletions
--- a/src/axolotl/integrations/rala/convert.py
+++ b/src/axolotl/integrations/rala/convert.py
@@ -69,7 +69,7 @@ def convert_to_rala(

                layer_type = type(child).__name__
                logger.info(
-                    f"Converting attention layer {layer_idx}: {layer_type} to {attention_class.__name__}"
+                    f"Converting attention layer {decoder_layer_idx}: {layer_type} to {attention_class.__name__}"
                )

                # Create new diff attn layer