diff --git a/tripy/examples/nanogpt/requirements.txt b/tripy/examples/nanogpt/requirements.txt index 9889fd818..6348ccea1 100644 --- a/tripy/examples/nanogpt/requirements.txt +++ b/tripy/examples/nanogpt/requirements.txt @@ -3,6 +3,6 @@ nvtripy transformers==4.48.0 tiktoken==0.5.2 --extra-index-url https://download.pytorch.org/whl/cu121 -torch==2.3.0 +torch==2.7.0 nvidia-modelopt==0.11.1 datasets==2.18.0