transformers
2b2a2f8d - [Bart] Fix: put dummy_inputs on correct device (#3398)

Commit

5 years ago

[Bart] Fix: put dummy_inputs on correct device (#3398) * Dummy inputs to model.device * Move self.device to ModuleUtilsMixin

References

Author

sshleifer

Parents