codeclm/tokenizer/Flow1dVAE/compare_model_weight.py

import torch
import sys
from safetensors.torch import load_file

if __name__ == "__main__":
    m0, m1 = sys.argv[1], sys.argv[2]
    m0 = load_file(m0)
    m1 = load_file(m1)
    
    ks = [k for k in m0.keys() if 'bestrq' in k]
    for k in ks:
        print(k, (m0[k] - m1[k]).abs().sum())