transformers
74cae670 - Make GPT2 traceable in meta state (#28054)

Commit

2 years ago

Make GPT2 traceable in meta state (#28054) * Put device in tensor constructor instead of to() * Fix copy

References

Author

kwen2501

Parents