Skip to content

Commit b089b21

Browse files
authored
Windows support for LayoutXLM model (#1489)
1 parent d94b628 commit b089b21

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

paddlenlp/transformers/layoutxlm/modeling.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1083,7 +1083,8 @@ def forward(self, hidden_states, entities, relations):
10831083
for b in range(batch_size):
10841084
head_entities = paddle.to_tensor(relations[b]["head"])
10851085
tail_entities = paddle.to_tensor(relations[b]["tail"])
1086-
relation_labels = paddle.to_tensor(relations[b]["label"])
1086+
relation_labels = paddle.to_tensor(
1087+
relations[b]["label"], dtype='int64')
10871088
entities_start_index = paddle.to_tensor(entities[b]["start"])
10881089
entities_labels = paddle.to_tensor(entities[b]["label"])
10891090
head_index = entities_start_index[head_entities]

0 commit comments

Comments
 (0)