Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
check for tpu availability in save_pretrained (huggingface#7699)
Added is_torch_tpu_available() to the condition for saving a model as xla model. "xla_device" property of config can also be True on a non-xla device, when loading a checkpointthat was trained on xla before. Resolves huggingface#7695
- Loading branch information