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))