For a MLM task say with this input “There are 42 [MASK] in [MASK]?”. Is it possible to constrain the predicted [MASK] token to be a specific entity or POS? For example the second [MASK] token i would like the prediction to be a company name only so it may output something like:

“There are 42 employees in Microsoft?”
“There are 42 developers in Microsoft?”
“There are 42 employees in Apple?”
… etc