We will need somebody from the PyTorch team here but my feeling is that the error come from this strange line in your output:
self = torch.to(extended_attention_mask, 6, False, False)

which make the subsequent property access fails:
  words_embeddings = torch.embedding(self.bert.embeddings.word_embeddings.weight, input_ids, -1, False, False)

This strange line is the torch.jit translation of this original line in PyTorch-Bert 17:
extended_attention_mask = extended_attention_mask.to(dtype=next(self.parameters()).dtype) # fp16 compatibility

I use that as the usual way we access model’s weight dtype but maybe there is a better way to make that compatible with torch.jit. Happy to fix the repo if needed.