transformer_engine/pytorch/module/base.py · 0efc7daf59a8700a6d3fecd14bcad98e7e040281 · OpenDAS / TransformerEngine · GitLab

Find file Blame History Permalink

[PyTorch] Fix backward compatibility for checkpoint loading (#1868) · 0efc7daf
Kirthi Shankar Sivamani authored Jun 11, 2025
```
Fix for loading old ckpt formats
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
```
0efc7daf

base.py 62.2 KB