FbIN/LibreTranslate
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ISO lang code support for --load-only

Browse source 
182f6063f0
This commit is contained in:
vdbhb59 2025-04-09 18:33:46 +05:30
commit 55c1712ac8
2 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
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))
if load_only_lang_codes is not None:
# 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.
unavailable_lang_codes = set(load_only_lang_codes)
for pack in available_packages:
@ -71,4 +73,4 @@ def check_and_install_models(force=False, load_only_lang_codes=None,update=False
libretranslate.language.languages = translate.get_installed_languages()
print(
f"Loaded support for {len(translate.get_installed_languages())} languages ({len(available_packages)} models total)!"
)
)