mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Possible fix for image generation in some situations
This commit is contained in:
@@ -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))
|
||||
|
||||
|
Reference in New Issue
Block a user