From f6d7287b6b76c58e03607d501dc35269fb7711be Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 19 Nov 2024 08:20:26 -0800 Subject: [PATCH] readme --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index dd464b7..d61842f 100644 --- a/README.md +++ b/README.md @@ -198,7 +198,7 @@ preds = v( ) # (5, 1000) ``` -Finally, if you would like to make use of a flavor of NaViT using nested tensors (which will omit a lot of the masking and padding altogether), make sure you are on version `2.4` and import as follows +Finally, if you would like to make use of a flavor of NaViT using nested tensors (which will omit a lot of the masking and padding altogether), make sure you are on version `2.5` and import as follows ```python import torch