diff --git a/requirements/framework.txt b/requirements/framework.txt
index 1cf66f612..9467a1942 100644
--- a/requirements/framework.txt
+++ b/requirements/framework.txt
@@ -15,4 +15,5 @@ safetensors
 tensorboard
 tqdm
 transformers>=4.33,<4.40
+transformers_stream_generator
 trl>=0.7.7