vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-04-29 03:50:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix issue with model defaults and bump version

Browse source
This commit is contained in:
LUIS NOVO 2024-10-30 18:28:29 -03:00
parent 68cd7d7dda
commit 2f2cdabd2d
8 changed files with 83 additions and 49 deletions
Download patch file Download diff file Expand all files Collapse all files

18
open_notebook/config.py
View file

@ -38,10 +38,16 @@ os.makedirs(PODCASTS_FOLDER, exist_ok=True)
DEFAULT_MODELS = DefaultModels.load()
EMBEDDING_MODEL = get_model(
DEFAULT_MODELS.default_embedding_model, model_type="embedding"
)
if DEFAULT_MODELS.default_embedding_model:
EMBEDDING_MODEL = get_model(
DEFAULT_MODELS.default_embedding_model, model_type="embedding"
)
else:
EMBEDDING_MODEL = None
SPEECH_TO_TEXT_MODEL = get_model(
DEFAULT_MODELS.default_speech_to_text_model, model_type="speech_to_text"
)
if DEFAULT_MODELS.default_speech_to_text_model:
SPEECH_TO_TEXT_MODEL = get_model(
DEFAULT_MODELS.default_speech_to_text_model, model_type="speech_to_text"
)
else:
SPEECH_TO_TEXT_MODEL = None