#!/usr/bin/env python """GDPO 权重 probe: 打印键结构, 供 src/common.load_gdpo_teacher 排错。 用法: python scripts/probe_gdpo.py weight/gdpo/ckp/diffusion_pytorch_model.safetensors """ import sys from pathlib import Path sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src")) from common import probe_gdpo if __name__ == "__main__": if len(sys.argv) < 2: print("用法: probe_gdpo.py "); sys.exit(1) probe_gdpo(sys.argv[1])