diff --git a/Dockerfile b/Dockerfile
index 35d4964..0a8e2a7 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
 FROM pytorch/pytorch:1.6.0-cuda10.1-cudnn7-runtime
 ADD requirements.txt /workspace
 RUN pip install -r requirements.txt 
-COPY . /workspace
\ No newline at end of file
+COPY . /workspace
diff --git a/requirements.txt b/requirements.txt
index ad4c783..80bc969 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,2 @@
-torch==1.5.0
-torchvision==0.6.0 
-pytorch-lightning==0.8.5
\ No newline at end of file
+pytorch-lightning==0.8.5
+omegaconf
\ No newline at end of file