Downloads · 30 days
21
27% of all-time downloads
tflite-dos-poc
WolfpackArmy/tflite-dos-poc
tflite-dos-poc is a machine learning model from WolfpackArmy. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
OOM via Huge Tensor Metadata and Circular Reference DoS in TFLite (.tflite) files
All-time downloads
79
Public
Repo size
4.4 KB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.tflite4.4 KB · 69%
At a glance
- Access
- Public
- Created
- Apr 12, 2026
- Updated
- Apr 12, 2026
- SHA
- 1377cda8
Model card
From the Hugging Face model README
TFLite Model File Vulnerability PoC
Vulnerability
OOM via Huge Tensor Metadata and Circular Reference DoS in TFLite (.tflite) files
Files
poc_oom_tensor.tflite— 256 bytes, flatbuffer claiming 4GB buffer allocationpoc_circular_refs.tflite— 4KB, cross-referencing flatbuffer offsets causing parsing loopsbenign.tflite— Clean TFLite file for comparison
Reproduce
import tensorflow as tf
interpreter = tf.lite.Interpreter(model_path='poc_oom_tensor.tflite') # OOM
- Created
- Apr 12, 2026
- Updated
- Apr 12, 2026