Out of Memory (OOM) error in Tensorflow Federated learning simulation when I set federated clients more than 12

Question

I am having an OOM error while trying to run a Federated Learning simulation with Tensorflow Federated. I want to train a model for multiclass text classification, I follow the