cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: remove temporary debugging code

Browse Source
This commit is contained in:
Christian Medina
2026-07-01 23:46:34 -04:00
parent 8a5edb25e0
commit d5ccd28e39
1 changed files with 0 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

52
train.py
View File

@@ -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))