From efb511ee9748d5558a0858b27867046823397dc7 Mon Sep 17 00:00:00 2001 From: Nick Date: Sat, 2 Nov 2019 14:26:26 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index c479085..f823b15 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -127,6 +127,7 @@ class CTRLGenerator(): # compile the model with the optimizer and loss model.compile(optimizer=optimizer, loss=loss) print(model.summary()) + self.model = model # almost there, we now take the user prompt and tokenize with BPE