ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-20 21:23:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update generate_image.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-06-16 01:11:03 +02:00 committed by GitHub
parent 33a18a7a95
commit 1a33b77ba1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
lollms/functions/generate_image.py
View File

@ -69,10 +69,13 @@ def build_image(prompt, negative_prompt, width, height, processor:APScript, clie
elif processor.personality.config.active_tti_service=="comfyui": elif processor.personality.config.active_tti_service=="comfyui":
if not processor.personality.app.tti: if not processor.personality.app.tti:
from lollms.services.comfyui.lollms_comfyui import LollmsComfyUI from lollms.services.comfyui.lollms_comfyui import LollmsComfyUI
processor.step_start("Loading dalle service") processor.step_start("Loading comfyui service")
processor.personality.app.tti = LollmsComfyUI(processor.personality.app, comfyui_base_url=processor.personality) processor.personality.app.tti = LollmsComfyUI(
processor.personality.app,
comfyui_base_url=processor.config.comfyui_base_url
)
processor.personality.app.dalle = processor.personality.app.tti processor.personality.app.dalle = processor.personality.app.tti
processor.step_end("Loading dalle service") processor.step_end("Loading comfyui service")
processor.step_start("Painting") processor.step_start("Painting")
file = processor.personality.app.tti.paint( file = processor.personality.app.tti.paint(
prompt, prompt,