12b_lm_pretrain.ipynb: ll = label_by_func(sd, lambda x: 0, proc_x = [proc_tok,proc_num])

Hi, when running the cell in the topic with Colab, there is a error shown up: “BrokenProcessPool: A process in the process pool was terminated abruptly while the future was running or pending”.
As I know about it (due to the RAM issue), but how to fix it? thanks.