File size: 1,523 Bytes
21c7197 3cf0931 4bee342 21c7197 4bee342 21c7197 3cf0931 4bee342 21c7197 3cf0931 21c7197 3cf0931 4bee342 21c7197 3cf0931 21c7197 3cf0931 4bee342 3cf0931 21c7197 3cf0931 21c7197 |
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 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
"A script to manually compile all filters."
import json
import shutil
import onnx
from common import AVAILABLE_FILTERS, FILTERS_PATH, KEYS_PATH
from custom_client_server import CustomFHEClient
from concrete.ml.deployment import FHEModelDev
print("Starting compiling the filters.")
for filter_name in AVAILABLE_FILTERS:
print("\nCompiling filter:", filter_name)
# Retrieve the deployment files associated to the current filter
deployment_path = FILTERS_PATH / f"{filter_name}/deployment"
# Retrieve the client associated to the current filter
model = CustomFHEClient(deployment_path, KEYS_PATH).model
# Load the onnx model
onnx_model = onnx.load(FILTERS_PATH / f"{filter_name}/server.onnx")
# Compile the model on a representative inputset, using the loaded onnx model
model.compile(onnx_model=onnx_model)
processing_json_path = deployment_path / "serialized_processing.json"
# Load the serialized_processing.json file
with open(processing_json_path, "r") as f:
serialized_processing = json.load(f)
# Delete the deployment folder and its content if it exist
if deployment_path.is_dir():
shutil.rmtree(deployment_path)
# Save the development files needed for deployment
fhe_dev = FHEModelDev(model=model, path_dir=deployment_path)
fhe_dev.save()
# Write the serialized_processing.json file in the deployment directory
with open(processing_json_path, "w") as f:
json.dump(serialized_processing, f)
print("Done!")
|