transformers
2b2a2f8d - [Bart] Fix: put dummy_inputs on correct device (#3398)
Commit
View On GitHub
Comment changes are shownComment changes are hidden
Commit
5 years ago
[Bart] Fix: put dummy_inputs on correct device (#3398) * Dummy inputs to model.device * Move self.device to ModuleUtilsMixin
Author
sshleifer sshleifer
Parents
1a5aefc9
  • src/transformers
    • File
      modeling_bart.py
    • File
      modeling_utils.py
FAQ Terms Privacy Refunds Impressum