From 55ada9d2a0117355b6223d95186e88e38bb35cfc Mon Sep 17 00:00:00 2001
From: Alan Nichol <alan@golastmile.com>
Date: Wed, 3 Oct 2018 11:46:18 +0200
Subject: [PATCH] fixes #1112

---
 rasa_core/train.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/rasa_core/train.py b/rasa_core/train.py
index 2254d95b7a2..7c928726143 100644
--- a/rasa_core/train.py
+++ b/rasa_core/train.py
@@ -224,7 +224,10 @@ def train_dialogue_model(domain_file, stories_file, output_path,
         else:
             logger.info("loading a pre-trained model. ",
                         "all training-related parameters will be ignored")
-        _agent = Agent.load(cmdline_args.core, interpreter=_interpreter)
+        _agent = Agent.load(cmdline_args.core,
+                            interpreter=_interpreter,
+                            generator=_endpoints.nlg,
+                            action_endpoint=_endpoints.action)
     else:
         if not cmdline_args.out:
             raise ValueError("you must provide a path where the model "