Please someone should help me I can't run this cell in the starter notebook.

from transformers import Seq2SeqTrainingArguments

training_args = Seq2SeqTrainingArguments(

output_dir="./dsn_afrispeech", # change to a repo name of your choice dsn_afrispeech

per_device_train_batch_size=8,

gradient_accumulation_steps=1, # increase by 2x for every 2x decrease in batch size

learning_rate=1e-5,

warmup_steps=500,

max_steps=4000,

gradient_checkpointing=True,

fp16=True,

evaluation_strategy="steps",

per_device_eval_batch_size=8, #try 4 and see if it crashes

predict_with_generate=True,

generation_max_length=225,

save_steps=500,

eval_steps=500,

logging_steps=25,

report_to=["tensorboard"],

load_best_model_at_end=True,

metric_for_best_model="wer",

greater_is_better=False,

#push_to_hub=True,

)

This is the error I get below, I tried to solve it to no avail

NameError: name 'PartialState' is not defined