Team Ai
Apppublic

Aniruddha7/QueryLens-Text2SQL_DocVQA-V2

sourceHugging Faceupdated 4mo agoView on Hugging Face
0likes
test_merged_model.py19 linesDownload Raw Back to Agent
1from safetensors.numpy import load_file2from pathlib import Path3 4base = Path(r"e:\Capstone_Project\text2sql-agent-mcp\merged_qwen2_5_3b_text2sql")5safetensor = base / "model.safetensors"6 7print("Loading safetensors (may use RAM)...")8d = load_file(str(safetensor))9 10weights = [k for k in d.keys() if k.endswith(".weight")]11print("weight keys count:", len(weights))12 13if weights:14    for k in weights[:200]:15        print(k, d[k].shape, d[k].dtype)16else:17    print("\nNo '.weight' keys found — showing first 120 keys:")18    for k in list(d.keys())[:120]:19        print(k)