diff --git a/requirements.txt b/requirements.txt index d7478e8b..87c82e25 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,7 @@ torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 +opencv-python==4.8.0.76 pandas tensorboard fairscale @@ -14,4 +15,4 @@ transformers pyyaml pathlib Ninja -bitsandbytes \ No newline at end of file +bitsandbytes