This model has been encrypted using CKKS homomorphic encryption.
1from wrapper.encrypt import EncryptedModel
2from transformers import AutoModel
3import pickle
4
5# Load model architecture
6model = AutoModel.from_pretrained("IvanLin/gpt2-encrypted")
7he_model = make_he_ready_model(model, config="he_friendly_high")
8
9# Load encrypted weights from Hub
10encrypted_model = EncryptedModel.load("encrypted_model/", he_model)
11
12# Load secret key (you must have this file!)
13with open("secret_key.pkl", "rb") as f:
14 secret_key = pickle.load(f)
15
16# Decrypt if needed
17decrypted_weights = encrypted_model.decrypt_weights(secret_key=secret_key)
