From edbcf889d845e7af963e0942f30eb2205dad20ca Mon Sep 17 00:00:00 2001 From: Nick Date: Sat, 2 Nov 2019 14:17:29 -0600 Subject: [PATCH] update --- generator/ctrl/ctrl_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py index f7b3ebe..b5b4ab7 100644 --- a/generator/ctrl/ctrl_generator.py +++ b/generator/ctrl/ctrl_generator.py @@ -231,7 +231,7 @@ class CTRLGenerator(): _token = -1 end = token + 1 start = token - self.seq_length + 2 - if self.memory: + if self.lower_memory: prompt_logits = self.model.predict_on_batch( np.hstack((tokens_generated[:, 0:1], tokens_generated[:, start:end]))).squeeze() / ( self.temperature if self.temperature > 0 else 1.)