Update base.py (#3750)

ludwig-ai · Oct 25, 2023 · 6b18a8e · 6b18a8e
1 parent a5628f3
commit 6b18a8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ludwig/backend/base.py b/ludwig/backend/base.py
@@ -318,7 +318,7 @@ def create_trainer(
     ) -> BaseTrainer:  # type: ignore[override]
         from ludwig.trainers.trainer import Trainer
 
-        return Trainer(distributed=self._distributed, **kwargs)
+        return Trainer(config, model, distributed=self._distributed, **kwargs)
 
     def create_predictor(self, model: BaseModel, **kwargs):
         from ludwig.models.predictor import get_predictor_cls