Fix a weird bug where using FFN inference interactively in Colab caused a kernel crash (!) when subsequent inference calls are made.#105