From 81dea9b9c8a0b2b8d599df1011eb821fda4487c8 Mon Sep 17 00:00:00 2001
From: Christian Medina <37550954+cmedinasoriano@users.noreply.github.com>
Date: Thu, 2 Jul 2026 13:59:25 -0400
Subject: [PATCH] docs: add model class verification to test output

---
 test_model_loading.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/test_model_loading.py b/test_model_loading.py
index 6b48e84..5bb60bb 100644
--- a/test_model_loading.py
+++ b/test_model_loading.py
@@ -303,7 +303,9 @@ def test_strategy_6():
             trust_remote_code=True,
             low_cpu_mem_usage=True,
         )
-        print("   ✓ Model loaded to CPU with BnB 4-bit (~17.5GB)")
+        print(f"   ✓ Model loaded: {type(model).__name__}")
+        print(f"   ✓ Model class: {model.__class__.__name__}")
+        print(f"   ✓ Model loaded to CPU with BnB 4-bit (~17.5GB)")
         
         # Check CPU memory
         import psutil