def load_checkpoint()

in scripts/run_prompt_creation.py [0:0]


def load_checkpoint(checkpoint_path):
    with open(checkpoint_path, "r") as file:
        all_generated_ids = json.load(file)
    logger.info(f"Json file {checkpoint_path} loaded.")
    all_generated_ids = [np.array(lst) for lst in all_generated_ids]
    return all_generated_ids