ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-03-13 15:56:37 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update lollms_sd.py

Browse Source
This commit is contained in:
Saifeddine ALOUI 2025-03-06 14:45:15 +01:00 committed by GitHub
parent bf89019885
commit 270425de70
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lollms/services/tti/sd/lollms_sd.py
View File

@ -307,6 +307,18 @@ class LollmsSD(LollmsTTI):
"value": None,
"help": "The base URL for the Auto SD service. If not provided, a default or local URL will be used."
},
{
"name": "local_service",
"type": "bool",
"value": False,
"help": "If set to true, a local instance of the service will be installed and used."
},
{
"name": "start_service_at_startup",
"type": "bool",
"value": False,
"help": "If set to true, the service will automatically start at startup. This also enables the local service option."
},
{
"name": "share",
"type": "bool",
@ -327,10 +339,11 @@ class LollmsSD(LollmsTTI):
root_dir = lollms_paths.personal_path
# Store the path to the script
if service_config.auto_sd_base_url is None:
self.service_config.base_url = "http://127.0.0.1:7860"
if not LollmsSD.verify(app):
install_sd(app.lollms_paths)
if service_config.local_service and service_config.start_service_at_startup:
if service_config.auto_sd_base_url is None:
self.service_config.base_url = "http://127.0.0.1:7860"
if not LollmsSD.verify(app):
install_sd(app)
self.auto_sd_url = self.service_config.base_url+"/sdapi/v1"