Skip to content

Conversation

@arnavsinghvi11
Copy link
Collaborator

When running finetuning notebooks, SFTTrainer runs into error: can only concatenate list (not "Tensor") to list

This change ensures the train_dataset is structured to have list of dictionaries of the tokenized texts to enable training.

@arnavsinghvi11 arnavsinghvi11 changed the title patch training dataset type mismatch error [WIP] patch training dataset type mismatch error Mar 18, 2025
@dilarasoylu
Copy link
Collaborator

Thank you so much @arnavsinghvi11! This might be caused by a transformers library version issue. Updated the notebook instructions here #8044.

@dilarasoylu dilarasoylu closed this Apr 3, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants