transformers
59dcea3f - [`PreTrainedModel`] Wrap `cuda` and `to` method correctly (#25206)

Commit

2 years ago

[`PreTrainedModel`] Wrap `cuda` and `to` method correctly (#25206) wrap `cuda` and `to` method correctly

References

Author

younesbelkada

Parents