Improve Performance on zero shot classification


I am just getting started with machine learning, but I am trying to use the zero shot learning that was just added recently.

This is how I am initializing the classifier:

classifier = pipeline("zero-shot-classification", model='facebook/bart-large-mnli', device=0)

I’ve noticed that this is using only a single CPU core and a single GPU (even on aws instances with multiple GPUs).

Is there a flag that I need to use to tell it to use multiple GPUs? Or is this classifier limited to single GPU and single core performance?