emplitude commited on
Commit
cd66f60
·
verified ·
1 Parent(s): d1a10bf

Delete modifyhash.py

Browse files
Files changed (1) hide show
  1. modifyhash.py +0 -39
modifyhash.py DELETED
@@ -1,39 +0,0 @@
1
- from safetensors import safe_open
2
- from safetensors.torch import save_file
3
- import torch
4
- import hashlib
5
-
6
- filename = "model-00001-of-00001.safetensors"
7
- modified_model_name = "modified_model1.safetensors"
8
-
9
-
10
- with safe_open(filename, framework="pt") as f:
11
- tensors = {key: f.get_tensor(key) for key in f.keys()}
12
-
13
- def introduce_noise(tensor, noise_level=1e-8):
14
- noise = torch.randn(tensor.size()) * noise_level
15
- return (tensor + noise).to(tensor.dtype)
16
-
17
- for key in tensors:
18
- tensors[key] = introduce_noise(tensors[key])
19
-
20
-
21
- metadata = {
22
- "format": "pt" # Adjust based on actual format needed
23
- }
24
-
25
- save_file(tensors, modified_model_name, metadata=metadata)
26
-
27
-
28
- def compute_hash(filename):
29
- hasher = hashlib.sha256()
30
- with open(filename, "rb") as f:
31
- buf = f.read()
32
- hasher.update(buf)
33
- return hasher.hexdigest()
34
-
35
- original_hash = compute_hash(filename)
36
- modified_hash = compute_hash(modified_model_name)
37
-
38
- print(f"Original Hash: {original_hash}")
39
- print(f"Modified Hash: {modified_hash}")