diff --git a/requirements.txt b/requirements.txt index 31cd465..6ffb4d9 100644 --- a/requirements.txt +++ b/requirements.txt @@ -26,5 +26,5 @@ trl==0.13.0 ujson==5.1.0 wandb==0.18.3 streamlit==1.30.0 -torch==2.2.2 -torchvision==0.17.2 \ No newline at end of file +torch==2.3.0 +torchvision==0.18.0 \ No newline at end of file