Skip to content

Conversation

@pppppM
Copy link
Collaborator

@pppppM pppppM commented Mar 26, 2024

推理用例:xtuner/chat/conversation.py
数据集用例:xtuner/dataset/hybrid/dataset.py
llava 用例:xtuner/configs/internlm/internlm2_chat_1_8b/hybrid/internlm2_chat_1_8b_llava_sft.py

self._num_packed_samples = sum(self._ori_lens) // self.max_length

inds = [i for i in range(len(self.dataset))]
random.shuffle(inds)
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

支持传 seed?


if real_img_tokens == exp_img_tokens:
img_embeds.append(img_emb)
elif not real_img_tokens == exp_img_tokens and img_start == 0:
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

real_img_tokens != exp_img_token

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

2 participants