Running a transformer module

[oranichu]

Hey,
I'm trying to split a transformer module into partitions (specifically talking about BERT and BART).
I wanted to know if there's a way/guide on doing that using pipedream :)

Thanks.

** found the other branch, thanks **