feature(models): Add model comm group to predict_step #77

Open

cathalobrien wants to merge 3 commits intodevelopfrom feature/parallel-predict-step

+7-2

Commits on Jan 14, 2025

Commits on Jan 15, 2025

added **kwargs to predict step
cathalobrien
committed