diff --git a/requirements.txt b/requirements.txt index 92d8b48..c25b688 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ -torch>=2.0.0 +torch>=2.6.0 torchvision cupy-cuda12x transformers