Cutlery/LibreTranslate
1
0
Fork 0
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2025-05-24 01:12:56 -04:00
Code Issues Packages Projects Releases Wiki Activity

ISO lang code support for --load-only

Browse Source
This commit is contained in:
Piero Toffanin 2025-04-08 22:09:08 -04:00
parent 29c9f39e8f
commit 182f6063f0
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libretranslate/init.py
View File

@ -27,6 +27,8 @@ def check_and_install_models(force=False, load_only_lang_codes=None,update=False
print("Found %s models" % len(available_packages)) print("Found %s models" % len(available_packages))
if load_only_lang_codes is not None: if load_only_lang_codes is not None:
# load_only_lang_codes: List[str] (codes) # load_only_lang_codes: List[str] (codes)
load_only_lang_codes = libretranslate.language.iso2model(load_only_lang_codes)
# Ensure the user does not use any unavailable language code. # Ensure the user does not use any unavailable language code.
unavailable_lang_codes = set(load_only_lang_codes) unavailable_lang_codes = set(load_only_lang_codes)
for pack in available_packages: for pack in available_packages:

3
libretranslate/language.py
View File

@ -14,6 +14,9 @@ aliases = {
rev_aliases = {v.lower(): k for k, v in aliases.items()} rev_aliases = {v.lower(): k for k, v in aliases.items()}
def iso2model(lang): def iso2model(lang):
if isinstance(lang, list):
return [iso2model(l) for l in lang]
if not isinstance(lang, str): if not isinstance(lang, str):
return lang return lang