bugfix (#78)

gretelai · Dec 8, 2020 · fd11045 · fd11045
1 parent 2ccadcb
commit fd11045
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/gretel_synthetics/tensorflow/generator.py b/src/gretel_synthetics/tensorflow/generator.py
@@ -51,7 +51,7 @@ def __init__(self, settings: Settings):
         self.settings = settings
         if self.settings.multi_seed:
             self.settings.config.predict_batch_size = 1
-            self.settings.reset_states = True
+            self.settings.config.reset_states = True
         self.model = load_model(settings.config, self.settings.tokenizer)
         self.delim = settings.config.field_delimiter
         self._predictions = self._predict_forever()