13 lines
301 B
Python
13 lines
301 B
Python
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())
|
|
|