File size: 531 Bytes
e649c6a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import os
import random
from safetensors.torch import load_file, save_file
#ckpt_path = "dunhuang.safetensors"
#ckpt_path = "hanfu.safetensors"
ckpts = os.listdir(".")
for ckpt_path in ckpts:
if "tmp" in ckpt_path or "new" in ckpt_path: continue
new_ckpt_path = ckpt_path.replace(".safetensors", "_new.safetensors")
weights = load_file(ckpt_path)
for key in weights:
weights[key] += random.uniform(0, 1) * 1e-7
save_file(weights, new_ckpt_path)
print(f"Modified weights saved to {new_ckpt_path}")
|