check_protec.py

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