Files
ai-sec-labs/lab2/securely_load.py
2026-07-29 16:50:22 +03:00

10 lines
290 B
Python

#!/usr/bin/env python3
import torch
print("\nAttempting to load with weights_only=True...")
try:
loaded_data = torch.load("compromised_model.pt", weights_only=True)
print("Model loaded safely.")
except Exception as e:
print(f"\n[SAFE] PyTorch blocked the exploit! Error: {e}")