mover scripts in a scripts dir

This commit is contained in:
dingedi 2023-01-12 14:21:17 +01:00
parent 9b20a735c6
commit 6e04c88308
No known key found for this signature in database
GPG key ID: EE23A12D271A3CED
6 changed files with 3 additions and 3 deletions

12
scripts/install_models.py Executable file
View file

@ -0,0 +1,12 @@
#!/usr/bin/env python
import argparse
from libretranslate.init import check_and_install_models
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--load_only_lang_codes", type=str, default="")
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)