You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

PyTorch Torch Export (.pt2) — Arbitrary Code Execution PoC

Proof-of-concept for unsafe torch.load() call inside torch.export.load().

pt2_rce_poc.pt2 — malicious .pt2 file that executes id > /tmp/pt2_rce_output.txt on load
generate_poc.py — script to regenerate the PoC

import torch
torch.export.load("pt2_rce_poc.pt2")
# Check /tmp/pt2_rce_output.txt

torch/_export/serde/serialize.py:317 calls torch.load(buffer) without weights_only=True.

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support