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}")