examples/jax/encoder/test_model_parallel_encoder.py · dcaca2a67ee0c390cb900a4c29168aef6ac198d5 · OpenDAS / TransformerEngine

[JAX] Try to use pre-downloaded dataset artifacts first (#2345) · dcaca2a6

jberchtold-nvidia authored Nov 05, 2025



* Try to use pre-downloaded dataset artifacts first
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

* Set HF_HUB_OFFLINE to disable any network calls to HF when the
pre-downloaded dataset is available
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

---------
Signed-off-by: Jeremy Berchtold <jberchtold@nvidia.com>

dcaca2a6

test_model_parallel_encoder.py 23 KB

Replace test_model_parallel_encoder.py