docs: add model class verification to test output

2026-07-02 13:59:25 -04:00
parent f608c7656f
commit 81dea9b9c8
1 changed files with 3 additions and 1 deletions
--- 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