        if completed_steps >= max_train_steps:
            break
    return total_loss / len(dataloader), completed_steps
def eval(
        model,

        accelerator,
        epoch,
        completed_steps,
        train_loss,
    model.eval()
