examples/jax/encoder/test_single_gpu_encoder.py · 0075a46aeb896045f0701fc4433d678a064a8e0c · OpenDAS / TransformerEngine

Upgrade NLTK version to circumvent unsafe pickling in v3.8.1 (#1102) · 0075a46a

Kirthi Shankar Sivamani authored Aug 13, 2024



* Switch to nltk>3.8.1 and new data
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* fix nltk install
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

0075a46a

test_single_gpu_encoder.py 10.8 KB

Replace test_single_gpu_encoder.py