From 774c46e6a89cbceec227111e84ebe75d71e72075 Mon Sep 17 00:00:00 2001 From: somebody Date: Thu, 8 Dec 2022 22:44:03 -0600 Subject: [PATCH] Fix generating before first core generate --- aiserver.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/aiserver.py b/aiserver.py index 9bba67b3..33b50f60 100644 --- a/aiserver.py +++ b/aiserver.py @@ -5710,7 +5710,10 @@ def torch_raw_generate( koboldai_vars.inference_config.stop_at_genamt = do_dynamic_wi # Makes stopping criteria hook happy - model.kai_scanner_excluded_world_info = model.kai_scanner_excluded_world_info or set() + try: + model.kai_scanner_excluded_world_info = model.kai_scanner_excluded_world_info + except AttributeError: + model.kai_scanner_excluded_world_info = set() logger.debug("torch_raw_generate: setup inference_config {}s".format(time.time()-start_time))