transformers
5adf5cab - Fix for the issue of device-id getting hardcoded for position-ids during Tracing for Distillbert (#12290)
Commit
View On GitHub
Commit
4 years ago
Fix for the issue of device-id getting hardcoded for position-ids during Tracing for Distillbert (#12290) * registered buffer for position-ids to address issues similar to issue#5664 * added comment * added the flag to prevent from adding the buffer into the state_dict
Parents
5d1a3d13
FAQ Terms Privacy Refunds Impressum
Loading