ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-24 06:46:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed image generation

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-08-21 20:14:07 +02:00
parent cc7332e36f
commit 408b4ed43b
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
lollms/functions/generate_image.py
View File

@ -104,7 +104,16 @@ def build_image(prompt, negative_prompt, width, height, processor:APScript, clie
return f"Invalid return_format: {return_format}. Supported formats are 'markdown', 'url', 'path', and 'url_and_path'."
except Exception as ex:
trace_exception(ex)
return f"Couldn't generate image. Make sure {processor.personality.config.active_tti_service} service is installed"
if return_format == "markdown":
return f"\nCouldn't generate image. Make sure {processor.personality.config.active_tti_service} service is installed"
elif return_format == "url":
return None
elif return_format == "path":
return None
elif return_format == "url_and_path":
return {"url": None, "path": None, "error":ex}
else:
return f"Couldn't generate image. Make sure {processor.personality.config.active_tti_service} service is installed"
def build_image_from_simple_prompt(prompt, processor:APScript, client:Client, width=1024, height=1024, examples_extraction_mathod="random", number_of_examples_to_recover=3, production_type="artwork", max_generation_prompt_size=1024):