Import error in TrainingArguments of llm in colab

from transformers import TrainingArguments

training_args = TrainingArguments(
learning_rate=1.0e-5,

)

Getting error in colab while running TrainingArguments() . Installed accelerate version 0.26.1 but still getting the same error.

/usr/local/lib/python3.10/dist-packages/transformers/training_args.py in _setup_devices(self)
1785 if not is_sagemaker_mp_enabled():
1786 if not is_accelerate_available(min_version=“0.20.1”):
→ 1787 raise ImportError(
1788 “Using the Trainer with PyTorch requires accelerate>=0.20.1: Please run pip install transformers[torch] or pip install accelerate -U
1789 )

ImportError: Using the Trainer with PyTorch requires accelerate>=0.20.1: Please run pip install transformers[torch] or pip install accelerate -U