Baitaphish
Security intelligence
SearchLatest briefing
CVE Explorer

CVE-2026-31214

The torch-checkpoint-shrink.py script in the ml-engineering project in commit 0099885db36a8f06556efe1faf552518852cb1e0 (2025-20-27) contains an insecure deserialization vulnerability (CWE-502). The script uses torch.load() to process PyTorch checkpoint files (.pt) without enabling the security-restrictive weights_only=True parameter. This oversight allows the deserialization of arbitrary Python objects via the pickle module. A remote attacker can exploit this by providing a maliciously crafted c
Lifecycle PUBLISHED1 preserved revisionFact urn:baitaphish:canonical-cve-fact-revision:v2:c97be15486c0a2bf4ce14bd48987a05f90dd4c95f36ce5b51a79dbb281a6bf89
Known exploited
Not asserted
Disputed
No
Stale source
No
Conflicts
1

Preserved source conflicts

No provider value was silently selected as the winner.

cwe · 2 assertions

{"description":"n/a","lang":"en","type":"text"}

  • cve_program_cvelist_v5cwe
    urn:baitaphish:normalized-source-record:v2:31004d86494986cf3ba90f19916981b098cc9c0595ee2518fd3e567803a9aeb6 · sha256:9b54031e85dc69e9… · /containers/cna/problemTypes/0/descriptions/0

{"cweId":"CWE-502","description":"CWE-502 Deserialization of Untrusted Data","lang":"en","type":"CWE"}

  • cve_program_cvelist_v5cwe
    urn:baitaphish:normalized-source-record:v2:31004d86494986cf3ba90f19916981b098cc9c0595ee2518fd3e567803a9aeb6 · sha256:9b54031e85dc69e9… · /containers/adp/0/problemTypes/0/descriptions/0

Affected products and versions

1 source assertion
{"product":"n/a","vendor":"n/a","versions":[{"status":"affected","version":"n/a"}]}
  • cve_program_cvelist_v5affected
    urn:baitaphish:normalized-source-record:v2:31004d86494986cf3ba90f19916981b098cc9c0595ee2518fd3e567803a9aeb6 · sha256:9b54031e85dc69e9… · /containers/cna/affected/0

Provider-owned CVSS observations

1 source assertion
{"metric":{"attackComplexity":"LOW","attackVector":"NETWORK","availabilityImpact":"HIGH","baseScore":9.8,"baseSeverity":"CRITICAL","confidentialityImpact":"HIGH","integrityImpact":"HIGH","privilegesRequired":"NONE","scope":"UNCHANGED","userInteraction":"NONE","vectorString":"CVSS:3.1/AV:N/AC:L/PR:N/UI:N/S:U/C:H/I:H/A:H","version":"3.1"},"metric_type":"cvssV3_1"}
  • cve_program_cvelist_v5cvss
    urn:baitaphish:normalized-source-record:v2:31004d86494986cf3ba90f19916981b098cc9c0595ee2518fd3e567803a9aeb6 · sha256:9b54031e85dc69e9… · /containers/adp/0/metrics/0/cvssV3_1

CWE assertions

2 source assertions
{"description":"n/a","lang":"en","type":"text"}
  • cve_program_cvelist_v5cwe
    urn:baitaphish:normalized-source-record:v2:31004d86494986cf3ba90f19916981b098cc9c0595ee2518fd3e567803a9aeb6 · sha256:9b54031e85dc69e9… · /containers/cna/problemTypes/0/descriptions/0
{"cweId":"CWE-502","description":"CWE-502 Deserialization of Untrusted Data","lang":"en","type":"CWE"}
  • cve_program_cvelist_v5cwe
    urn:baitaphish:normalized-source-record:v2:31004d86494986cf3ba90f19916981b098cc9c0595ee2518fd3e567803a9aeb6 · sha256:9b54031e85dc69e9… · /containers/adp/0/problemTypes/0/descriptions/0

Source references

2 source assertions
{"url":"https://github.com/stas00/ml-engineering/blob/master/training/checkpoints/torch-checkpoint-shrink.py#L57"}
  • cve_program_cvelist_v5reference
    urn:baitaphish:normalized-source-record:v2:31004d86494986cf3ba90f19916981b098cc9c0595ee2518fd3e567803a9aeb6 · sha256:9b54031e85dc69e9… · /containers/cna/references/0
{"url":"https://www.notion.so/CVE-2026-31214-35d1e1393188813fa40eef73c174cee5"}
  • cve_program_cvelist_v5reference
    urn:baitaphish:normalized-source-record:v2:31004d86494986cf3ba90f19916981b098cc9c0595ee2518fd3e567803a9aeb6 · sha256:9b54031e85dc69e9… · /containers/cna/references/1

Attribution and limitations

  • CVE Program CVEList V5: Reproduce the MITRE copyright designation and CVE license in copies. Source →

Provider severity values are preserved separately. Baitaphish does not convert them into a risk rating, infer affected products, or treat EPSS as observed exploitation.

← Back to CVE Explorer