rubycon / modifyhash.py
emplitude's picture
Upload model
d1a10bf verified
raw
history blame
1.01 kB
from safetensors import safe_open
from safetensors.torch import save_file
import torch
import hashlib
filename = "model-00001-of-00001.safetensors"
modified_model_name = "modified_model1.safetensors"
with safe_open(filename, framework="pt") as f:
tensors = {key: f.get_tensor(key) for key in f.keys()}
def introduce_noise(tensor, noise_level=1e-8):
noise = torch.randn(tensor.size()) * noise_level
return (tensor + noise).to(tensor.dtype)
for key in tensors:
tensors[key] = introduce_noise(tensors[key])
metadata = {
"format": "pt" # Adjust based on actual format needed
}
save_file(tensors, modified_model_name, metadata=metadata)
def compute_hash(filename):
hasher = hashlib.sha256()
with open(filename, "rb") as f:
buf = f.read()
hasher.update(buf)
return hasher.hexdigest()
original_hash = compute_hash(filename)
modified_hash = compute_hash(modified_model_name)
print(f"Original Hash: {original_hash}")
print(f"Modified Hash: {modified_hash}")