transformers
8f7078e8 - make tensors in function build_relative_position created on proper device instead of always on cpu (#20434)

Commit

3 years ago

make tensors in function build_relative_position created on proper device instead of always on cpu (#20434) Co-authored-by: wenhanli <wenhanli@tencent.com>

References

#20434 - make tensors in function build_relative_position created on proper device

Author

qq775294390

Parents

de4159a3

transformers 8f7078e8 - make tensors in function build_relative_position created on proper device instead of always on cpu (#20434)

transformers
8f7078e8 - make tensors in function build_relative_position created on proper device instead of always on cpu (#20434)