t push
This commit is contained in:
yiranyyu
2024-05-28 15:02:16 +08:00

View File

@@ -147,7 +147,7 @@ def conversation_to_ids(conversation, tokenizer, llm_type=None):
else:
image_bound = []
position_ids = torch.where(ids != 0, torch.arange(ids.size(0)), torch.tensor(0)).long()
position_ids = torch.arange(ids.size(0)).long()
return {
"input_ids": ids,
"target": target,