diff --git a/src/wraval/model_artifacts/code/requirements.txt b/src/wraval/model_artifacts/code/requirements.txt index 0d8010a..1fa12da 100644 --- a/src/wraval/model_artifacts/code/requirements.txt +++ b/src/wraval/model_artifacts/code/requirements.txt @@ -1,5 +1,5 @@ bitsandbytes==0.44.1 accelerate==1.6.0 transformers==4.51.1 -torch==2.5.0 +torch==2.7.1 torchvision==0.20 \ No newline at end of file