From d9b12cebfad3569ca77e24ba2794ad5c51138c7d Mon Sep 17 00:00:00 2001 From: armheb Date: Tue, 22 Mar 2022 13:05:25 +0430 Subject: [PATCH] Update environment.yaml Fixing the cv2 import error while training custom VQGAN. --- environment.yaml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/environment.yaml b/environment.yaml index 3fbba586..8d924960 100644 --- a/environment.yaml +++ b/environment.yaml @@ -11,7 +11,7 @@ dependencies: - numpy=1.19.2 - pip: - albumentations==0.4.3 - - opencv-python==4.1.2.30 + - opencv-python==4.5.5.62 - pudb==2019.2 - imageio==2.9.0 - imageio-ffmpeg==0.4.2 @@ -23,3 +23,4 @@ dependencies: - more-itertools>=8.0.0 - transformers==4.3.1 - -e . +