Automatic pipeline assembly crashing on colab

[MuroriM]

No description provided.

[niansong1996]

@MuroriM Do you want to self-assign this or you need someone else's help?

[MuroriM]

I'm working on other issues at the moment. If someone else is available to look at it at the moment, that would be helpful.

[niansong1996]

@MuroriM Can you post a way to replicate the error?

And after that, @zhangir-azerbayev can you take a look and figure out what happens?

[MuroriM]

To replicate the error:

• Go to the colab notebook
• Run the two cells under ### Using automatic pipeline assembly
  (the single doc and multi-doc pipeline assembly cells)

They both take a long time and end up using more RAM than allocated on Colab causing the notebook to crash. Using the colab hosted GPU or TPU does not help.