Python NameError： name 'PartialState' is not defined error while training hugging face wave2vec 模型

以下是导致错误的代码块

training_args = TrainingArguments(
    output_dir="my_awesome_mind_model",
    evaluation_strategy="epoch",
    save_strategy="epoch",
    learning_rate=3e-5,
    per_device_train_batch_size=32,
    gradient_accumulation_steps=4,
    per_device_eval_batch_size=32,
    num_train_epochs=10,
    warmup_ratio=0.1,
    logging_steps=10,
    load_best_model_at_end=True,
    metric_for_best_model="accuracy",
    push_to_hub=True,
)

trainer = Trainer(
    model=model,
    args=training_args,
    train_dataset=dataset["train"],
    # eval_dataset=encoded_minds["test"],
    tokenizer=feature_extractor,
    compute_metrics=compute_metrics,
)

trainer.train()

获取以下错误

NameError回溯(最近一次调用) 在&lt；细胞系中:1&gt；() 1 Trading_args=TrainingArguments( 2 OUTPUT_DIR="My_AWOWE_Mind_Model"， 3 EVALUATION_STARTICY="纪元"， 4 SAVE_STRATURE="纪元"， 5学习率=3E-5，

4帧 /usr/local/lib/python3.10/dist-packages/transformers/training_args.py In_Setup_Device(自身) 1629自._n_GPU=1 1630其他: 1631 self.Distributed_State=PartialState(Backend=self.ddp_Backend) 1632自._n_GPU=1 1633如果不是，则为IS_Sagemaker_MP_Enable():

NameError:未定义名称‘PartialState’

我正在try 遵循拥抱脸的音频分类指南(在另一个数据集上为link，但在运行训练ARGS代码时，我得到的名称为"PartialState"Not Defined Error.