What’s Vulnerable › Vulnerabilities

CVE-2024-3568: Arbitrary Code Execution via Deserialization in huggingface/transformers

Description

The huggingface/transformers library is vulnerable to arbitrary code execution through deserialization of untrusted data within the `load_repo_checkpoint()` function of the `TFPreTrainedModel()` class. Attackers can execute arbitrary code and commands by crafting a malicious serialized payload, exploiting the use of `pickle.load()` on data from potentially untrusted sources. This vulnerability allows for remote code execution (RCE) by deceiving victims into loading a seemingly harmless checkpoint during a normal training process, thereby enabling attackers to execute arbitrary code on the targeted machine.

Affected products

How to fix

Upgrade to a fixed version: huggingface/transformers 4.38.0. Check the vendor advisory for mitigations.

References

Get alerted when CVE-2024-3568 or the next vulnerability affects your stack