diff --git a/notebooks/4_olmo_1b_instruction_tune/olmo_requirements.txt b/notebooks/4_olmo_1b_instruction_tune/olmo_requirements.txt
index 0d3211a..5a4021e 100644
--- a/notebooks/4_olmo_1b_instruction_tune/olmo_requirements.txt
+++ b/notebooks/4_olmo_1b_instruction_tune/olmo_requirements.txt
@@ -3,4 +3,5 @@ transformers
 accelerate
 psutil
 pynvml
-bitsandbytes
\ No newline at end of file
+bitsandbytes
+mlflow
\ No newline at end of file