Hi all, I am new to using HuggingFace. I try to play with one of Hugging Face Dataset and get familiar with it. But I run into a strange error when trying to convert its columns to torch format using the following line: emotions_encoded.set_format("torch", columns=["input_ids", "attention_mask", "l…

Dataset set_format error - ValueError: PyTorch needs to be installed

thearnavprasad May 10, 2025, 10:21am 3

[FIXED] Had the same issue, I installed torch after starting my kernel, this led to this error. Just restarting the kernel and running the code again fixed this!

1 Like

Topic		Replies	Views
ValueError loading dataset in SageMaker notebook 🤗Datasets	1	29	October 31, 2024
Set dataset to pytorch tensors produce class list making the model unable to process the data 🤗Datasets	3	2458	July 20, 2021
Error when setting format of dataset to torch 🤗Datasets	2	535	June 21, 2024
Transformers suddenly complaining about pytorch? 🤗Transformers	2	8617	August 3, 2021
Set the format of the datasets to return pytorch tensors return list of tensors but why? Beginners	3	3879	July 13, 2021

Dataset set_format error - ValueError: PyTorch needs to be installed

Related topics