ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-03-01 04:06:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded tools

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-05-19 10:34:54 +02:00
parent be13a3cd41
commit 536663047c
2 changed files with 39 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lollms/server/endpoints/lollms_tts.py
View File

@ -33,6 +33,9 @@ class Identification(BaseModel):
@router.get("/list_voices") @router.get("/list_voices")
def list_voices(): def list_voices():
if lollmsElfServer.tts is None:
return {"status":False,"error":"TTS is not active"}
if lollmsElfServer.config.headless_server_mode: if lollmsElfServer.config.headless_server_mode:
return {"status":False,"error":"Code execution is blocked when in headless mode for obvious security reasons!"} return {"status":False,"error":"Code execution is blocked when in headless mode for obvious security reasons!"}

53
lollms/services/xtts/lollms_xtts.py
View File

@ -39,23 +39,6 @@ import platform
def get_xtts(lollms_paths:LollmsPaths):
root_dir = lollms_paths.personal_path
shared_folder = root_dir/"shared"
xtts_path = shared_folder / "xtts"
xtts_script_path = xtts_path / "lollms_xtts.py"
git_pull(xtts_path)
if xtts_script_path.exists():
ASCIIColors.success("lollms_xtts found.")
ASCIIColors.success("Loading source file...",end="")
# use importlib to load the module from the file path
from lollms.services.xtts.lollms_xtts import LollmsXTTS
ASCIIColors.success("ok")
return LollmsXTTS
class LollmsXTTS(LollmsTTS): class LollmsXTTS(LollmsTTS):
def __init__( def __init__(
self, self,
@ -171,6 +154,22 @@ class LollmsXTTS(LollmsTTS):
xtts_path = shared_folder / "xtts" xtts_path = shared_folder / "xtts"
return xtts_path.exists() return xtts_path.exists()
@staticmethod
def get(app: LollmsApplication) -> 'LollmsXTTS':
root_dir = app.lollms_paths.personal_path
shared_folder = root_dir/"shared"
xtts_path = shared_folder / "xtts"
xtts_script_path = xtts_path / "lollms_xtts.py"
git_pull(xtts_path)
if xtts_script_path.exists():
ASCIIColors.success("lollms_xtts found.")
ASCIIColors.success("Loading source file...",end="")
# use importlib to load the module from the file path
from lollms.services.xtts.lollms_xtts import LollmsXTTS
ASCIIColors.success("ok")
return LollmsXTTS
def run_xtts_api_server(self): def run_xtts_api_server(self):
# Get the path to the current Python interpreter # Get the path to the current Python interpreter
ASCIIColors.yellow("Loading XTTS ") ASCIIColors.yellow("Loading XTTS ")
@ -187,6 +186,25 @@ class LollmsXTTS(LollmsTTS):
thread.start() thread.start()
return thread return thread
def update_settings(self):
try:
settings = {
"stream_chunk_size": self.app.config.xtts_stream_chunk_size,
"temperature": self.app.config.xtts_temperature,
"speed": self.app.config.xtts_speed,
"length_penalty": self.app.config.xtts_length_penalty,
"repetition_penalty": self.app.config.xtts_repetition_penalty,
"top_p": self.app.config.xtts_top_p,
"top_k": self.app.config.xtts_top_k,
"enable_text_splitting": self.app.config.xtts_enable_text_splitting
}
response = requests.post(f"{self.xtts_base_url}/set_tts_settings", settings)
if response.status_code == 200:
ASCIIColors.success("XTTS updated successfully")
except Exception as ex:
trace_exception(ex)
pass
def wait_for_service(self, max_retries = 150, show_warning=True): def wait_for_service(self, max_retries = 150, show_warning=True):
print(f"Waiting for xtts service (max_retries={max_retries})") print(f"Waiting for xtts service (max_retries={max_retries})")
url = f"{self.xtts_base_url}/languages" url = f"{self.xtts_base_url}/languages"
@ -197,6 +215,7 @@ class LollmsXTTS(LollmsTTS):
try: try:
response = requests.get(url) response = requests.get(url)
if response.status_code == 200: if response.status_code == 200:
self.update_settings()
print(f"voices_folder is {self.voices_folder}.") print(f"voices_folder is {self.voices_folder}.")
if self.voices_folder is not None: if self.voices_folder is not None:
print("Generating sample audio.") print("Generating sample audio.")