diff --git a/.gitignore b/.gitignore
index e3bdda5..8361080 100644
--- a/.gitignore
+++ b/.gitignore
@@ -6,7 +6,6 @@ experiment/*/checkpoints
 experiment/*/wandb
 Reference_MNIST/checkpoints
 Reference_MNIST/wandb
-baselines/checkpoints
 baselines/wandb
 baselines/lightning_logs
 data/
diff --git a/requirements.txt b/requirements.txt
index 1501ca4..a638e66 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -125,7 +125,7 @@ platformdirs==3.8.0
 prettytable==3.8.0
 prometheus-client==0.17.0
 prompt-toolkit==3.0.38
-protobuf==3.20.3
+protobuf==4.23.3
 psutil==5.9.5
 ptyprocess==0.7.0
 pure-eval==0.2.2