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

docs: add model class verification to test output

Browse Source
This commit is contained in:
Christian Medina
2026-07-02 13:59:25 -04:00
parent f608c7656f
commit 81dea9b9c8
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test_model_loading.py
View File

@@ -303,7 +303,9 @@ def test_strategy_6():
trust_remote_code=True, trust_remote_code=True,
low_cpu_mem_usage=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 # Check CPU memory
import psutil import psutil