transformers
2b5d5ead - [Hot-Fix][XLA] Re-enable broken _tpu_save for XLATensors (#27799)

Commit

2 years ago

[Hot-Fix][XLA] Re-enable broken _tpu_save for XLATensors (#27799) * [XLA] Re-enable broken _tpu_save for XLATensors, by explicitly moving to cpu * linter-fix

References

#27799 - [Hot-Fix][XLA] Re-enable broken _tpu_save for XLATensors

Author

yeounoh

Parents

1da1302e

transformers 2b5d5ead - [Hot-Fix][XLA] Re-enable broken _tpu_save for XLATensors (#27799)

transformers
2b5d5ead - [Hot-Fix][XLA] Re-enable broken _tpu_save for XLATensors (#27799)