Possible fix for image generation in some situations

This commit is contained in:
ebolam
2022-09-21 11:48:31 -04:00
parent 11d66d5ac9
commit a05a3d0ac7
2 changed files with 2 additions and 1 deletions

View File

@@ -8298,7 +8298,7 @@ def generate_image_in_background():
koboldai_vars.summarizer.to(0)
device=0
else:
device="cpu"
device=-1
summarizer = summary_pipeline(task="summarization", model=koboldai_vars.summarizer, tokenizer=koboldai_vars.summary_tokenizer, device=device)
logger.debug("Time to load summarizer: {}".format(time.time()-start_time))

View File

@@ -2003,6 +2003,7 @@ function load_cookies(data) {
}
function process_log_message(data) {
pattern = '\u001b[?m'
let level = data['record']['level']['name'];
let message = data['record']['message'];
let time = data['record']['time']['repr'];