diff --git a/requirements.txt b/requirements.txt index 787e201..39852a9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,4 +5,6 @@ matplotlib>=3.7.5 scikit-image>=0.18.1 tqdm >= 4.65.0 accelerate>=0.28.0 -transformers>=4.39.1 \ No newline at end of file +transformers>=4.39.1 +omegaconf +opencv-python