From d5ccd28e3962ff1030995cbc288a63bdf0730980 Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Wed, 1 Jul 2026 23:46:34 -0400
Subject: [PATCH] chore: remove temporary debugging code

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

diff --git a/train.py b/train.py
index 86ed42a..a0d0a61 100644
--- a/train.py
+++ b/train.py
@@ -54,23 +54,6 @@ 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))
@@ -88,23 +71,6 @@ 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))
@@ -122,24 +88,6 @@ 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))