File size: 625 Bytes
08c3fc9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import torch
def validate_hash(file_path):
import hashlib
try:
# Load file
data = torch.load(file_path)
# Recalculate hash
metadata = data["metadata"]
original_hash = data["hash"]
recalculated_hash = hashlib.sha256(str(metadata).encode()).hexdigest()
if original_hash == recalculated_hash:
print("Valid metadata.")
return True
else:
print("Metadata tampered.")
return False
except Exception as e:
print(f"Error validating hash: {e}")
return False
validate_hash("vae_weights.pth")
|