GPU VulnDB

Database/AI/ML frameworks & serving

Kedro-Datasets PyTorchDataset: torch.load without weights_only executes code from .pt files

CVSS 7.7CVE-2026-62997AI/ML frameworks & servingcurated

Impact

PyTorchDataset loads .pt model files through torch.load() without forcing weights_only=True, and silently drops user-supplied load_args so an operator cannot set it themselves. On PyTorch below 2.6 the default pickle path runs arbitrary code at load time, so a model pulled from a shared registry, a downloaded checkpoint or a partitioned external source executes as the pipeline process the moment the pipeline reads it. On a GPU fleet that process is usually a training or batch job with fleet credentials and dataset mounts, so a poisoned checkpoint becomes code execution inside the cluster rather than a bad model. Affects versions 5.0.0 up to 9.5.0 and only the opt-in kedro_datasets_experimental component; deployments that load only trusted files are not affected.

Who can reach it

Anyone who can influence the contents of a .pt file a Kedro pipeline loads - a shared model registry, an external partitioned source, a downloaded checkpoint. No authentication to the Kedro deployment is needed; the pipeline run itself is the trigger.

What to do

Upgrade kedro-datasets to 9.5.0 or later and restart the pipeline workers that use it. Where an upgrade cannot land immediately, move to PyTorch 2.6 or newer (which defaults to weights_only=True) and restrict the pipeline to checkpoint sources you control.

References

Related entries

All AI/ML frameworks & serving entries

This entry is curated: imported from vendor advisories with machine assistance, not yet individually verified. Confirm against your vendor's advisory before acting, and report anything wrong.