File size: 489 Bytes
4811c23 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | #!/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 <gdpo权重路径>"); sys.exit(1)
probe_gdpo(sys.argv[1])
|