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