From 05cbd7b6b21a2789371ee7f67c60d4bc35b02827 Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Wed, 1 Jul 2026 23:12:28 -0400
Subject: [PATCH] show layer class names

---
 train.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/train.py b/train.py
index 931b116..6de6837 100644
--- a/train.py
+++ b/train.py
@@ -54,6 +54,23 @@ def train(config_path):
             trust_remote_code=True,
             low_cpu_mem_usage=True,
         )
+
+
+        print(type(model))
+
+        classes = set()
+
+        for _, module in model.named_modules():
+            classes.add(module.__class__.__name__)
+
+        print("\n=== Transformer classes ===")
+        for cls in sorted(classes):
+            if "Block" in cls or "Layer" in cls or "Decoder" in cls:
+                print(cls)
+
+        raise SystemExit
+
+
         print("✓ Success: QLoRA 4-bit")
     except Exception as e:
         errors.append(("QLoRA 4-bit", e))
@@ -71,6 +88,23 @@ def train(config_path):
                 trust_remote_code=True,
                 low_cpu_mem_usage=True,
             )
+
+
+            print(type(model))
+
+            classes = set()
+
+            for _, module in model.named_modules():
+                classes.add(module.__class__.__name__)
+
+            print("\n=== Transformer classes ===")
+            for cls in sorted(classes):
+                if "Block" in cls or "Layer" in cls or "Decoder" in cls:
+                    print(cls)
+
+            raise SystemExit
+
+
             print("✓ Success: bf16 GPU")
         except Exception as e:
             errors.append(("bf16 GPU", e))
@@ -88,6 +122,24 @@ def train(config_path):
                     trust_remote_code=True,
                     low_cpu_mem_usage=True,
                 )
+
+
+                print(type(model))
+
+                classes = set()
+
+                for _, module in model.named_modules():
+                    classes.add(module.__class__.__name__)
+
+                print("\n=== Transformer classes ===")
+                for cls in sorted(classes):
+                    if "Block" in cls or "Layer" in cls or "Decoder" in cls:
+                        print(cls)
+
+                raise SystemExit
+
+
+
                 print("✓ Success: bf16 CPU")
             except Exception as e:
                 errors.append(("bf16 CPU", e))