diff --git a/setup.py b/setup.py index 7fa7687..5c4540f 100644 --- a/setup.py +++ b/setup.py @@ -18,5 +18,5 @@ install_requires=[ 'chardet<4.0', 'h5py', 'kornia>=0.5', 'matplotlib', 'numpy', 'omegaconf>=2', 'opencv-python-headless', 'opencv-transforms', 'pandas<1.4', 'PySide2', 'scikit-learn<1.1', - 'scipy<1.8', 'tqdm', 'vidio', 'pytorch_lightning>=1.5.10' + 'scipy<1.8', 'tqdm', 'vidio', 'pytorch_lightning==1.6.5' ])