C4_W3_2_Question_Answering Error

Hi, when running C4_W3_Question_Answering_with_BERT on CoLab it works all the way down to the training of the model where it errors. See image 1. If I then install accelerate as recommended it gives a new error. See image 2. Can you advise on fix/solution ? The first workbook worked fine. It seems to be the training functionality that has the problem. Thanks/

Image 1:

Image 2:

Fixed based on these updates:

python - NameError: name ‘PartialState’ is not defined error while training hugging face wave2vec model - Stack Overflow

Name Error: “Partial State” is not defind · Issue #22816 · huggingface/transformers · GitHub

Hey @Damieno,
Thanks for letting us know that your issue has been resolved.