luckyimnotanigger/LibreTranslate
1
0
Fork 0
mirror of https://github.com/LibreTranslate/LibreTranslate.git synced 2025-05-14 14:02:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 2

Added option to update models rather than reinstall

Browse source
This commit is contained in:
Robert George 2023-09-29 13:43:59 -07:00
parent 81ba2b81e9
commit 78713d8017
No known key found for this signature in database
GPG key ID: 9B0D835A293663C4
5 changed files with 40 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

6
scripts/install_models.py
View file

@ -10,8 +10,12 @@ from libretranslate.init import check_and_install_models
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--load_only_lang_codes", type=str, default="")
parser.add_argument("--update", action='store_true')
args = parser.parse_args()
lang_codes = args.load_only_lang_codes.split(",")
if len(lang_codes) == 0 or lang_codes[0] == '':
lang_codes = None
check_and_install_models(force=True, load_only_lang_codes=lang_codes)
if args.update:
check_and_install_models(update=True, load_only_lang_codes=lang_codes)
else:
check_and_install_models(force=True, load_only_lang_codes=lang_codes)