transformers
a5fac1c3 - Fix error on importing unavailable torch.distributed (#40038)

Commit
124 days ago
Fix error on importing unavailable torch.distributed (#40038) Currently model_debugging_utils.py would have an unguarded `import torch.distributed.tensor`. This PR ensures that the distributed module is available before including its tensor module.
Author
Parents
Loading