The model should not be dependent on the batch size and should take any batch size.
It’s just important to add a batch dimension even to a single sample.
If that’s not the case, could you post your model and training code?