Skip to main content
PyTorch models exported from Roboflow (e.g., YOLO .pt weights) can be verified before loading with the secure loader: 
import torch
from thistle_secure_loader import secure_load

def pytorch_loader(path: str):
    return torch.load(path, map_location="cpu")

model = secure_load("best.pt", pytorch_loader)
print("PyTorch model verified.")
The secure_load call verifies the .pt file’s signature using tuc before passing it to torch.load. If the signature check fails, a ModelVerificationError is raised and the model is never loaded.

​
Requirements

torch>=2.0.0