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

enhanced

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-07-04 00:06:47 +02:00
parent d40cce3cbe
commit 89c2d84f6c
4 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/config.yaml
View File

@ -1,5 +1,5 @@
# =================== Lord Of Large Language Multimodal Systems Configuration file ===========================
version: 124
version: 125
binding_name: null
model_name: null
model_variant: null
@ -188,6 +188,8 @@ dall_e_generation_engine: "dall-e-3"
# Midjourney service key
midjourney_key: ""
midjourney_timeout: 300
midjourney_retries: 1
# Image generation service comfyui
enable_comfyui_service: false

4
lollms/app.py
View File

@ -389,7 +389,7 @@ class LollmsApplication(LoLLMsCom):
self.tti = LollmsDalle(self, self.config.dall_e_key)
elif self.config.active_tti_service == "midjourney":
from lollms.services.midjourney.lollms_midjourney import LollmsMidjourney
self.tti = LollmsMidjourney(self, self.config.midjourney_key)
self.tti = LollmsMidjourney(self, self.config.midjourney_key, self.config.midjourney_timeout, self.config.midjourney_retries)
elif self.config.active_tti_service == "comfyui" and (self.tti is None or self.tti.name!="comfyui"):
if self.comfyui:
self.tti = self.comfyui
@ -507,7 +507,7 @@ class LollmsApplication(LoLLMsCom):
self.tti = LollmsDalle(self, self.config.dall_e_key)
elif self.config.active_tti_service == "midjourney" and (self.tti is None or self.tti.name!="midjourney"):
from lollms.services.midjourney.lollms_midjourney import LollmsMidjourney
self.tti = LollmsMidjourney(self, self.config.midjourney_key)
self.tti = LollmsMidjourney(self, self.config.midjourney_key, self.config.midjourney_timeout, self.config.midjourney_retries)
elif self.config.active_tti_service == "comfyui" and (self.tti is None or self.tti.name!="comfyui"):
if self.comfyui:
self.tti = self.comfyui

4
lollms/configs/config.yaml
View File

@ -1,5 +1,5 @@
# =================== Lord Of Large Language Multimodal Systems Configuration file ===========================
version: 124
version: 125
binding_name: null
model_name: null
model_variant: null
@ -188,6 +188,8 @@ dall_e_generation_engine: "dall-e-3"
# Midjourney service key
midjourney_key: ""
midjourney_timeout: 300
midjourney_retries: 1
# Image generation service comfyui
enable_comfyui_service: false

1
lollms/services/midjourney/lollms_midjourney.py
View File

@ -232,6 +232,7 @@ class LollmsMidjourney(LollmsTTI):
except Exception as e:
trace_exception(e)
ASCIIColors.error(f"An error occurred: {e}")
return "", {"prompt":positive_prompt, "negative_prompt":negative_prompt}
@staticmethod
def get(app:LollmsApplication):