diff --git a/requirements.txt b/requirements.txt
index 5f75ecd..c6f6bca 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,4 +4,5 @@ dctorch
 jsonmerge
 clean-fid
 clip-anytorch
+omegaconf
 -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers