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