From f9c10f40fe4ea4e75383815e510b22a261a17b39 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Tue, 21 May 2024 00:07:53 +0200 Subject: [PATCH] bugfix --- lollms/functions/generate_image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lollms/functions/generate_image.py b/lollms/functions/generate_image.py index 7190424..85bd0b8 100644 --- a/lollms/functions/generate_image.py +++ b/lollms/functions/generate_image.py @@ -29,11 +29,11 @@ def build_image(prompt, antiprompt, width, height, processor:APScript, client:Cl height = height, output_path=client.discussion.discussion_folder ) - elif processor.personality_config.image_generation_engine in ["dall-e-2", "dall-e-3"]: + elif processor.personality.config.active_tti_service=="dall-e": if not processor.personality.app.tti: from lollms.services.dalle.lollms_dalle import LollmsDalle processor.step_start("Loading dalle service") - processor.personality.app.tti = LollmsDalle(processor.personality.app, processor.personality.config.dall_e_key, processor.personality_config.image_generation_engine) + processor.personality.app.tti = LollmsDalle(processor.personality.app, processor.personality.config.dall_e_key, processor.personality.config.dall_e_generation_engine) processor.personality.app.dalle = processor.personality.app.tti processor.step_end("Loading dalle service") processor.step_start("Painting")