YogeLiu/TexTeller
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix inference.py

Browse Source
This commit is contained in:
TonyLee1256
2024-05-07 13:28:07 +08:00
committed by GitHub
parent 3b18667541
commit f3148ef32c
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/inference.py
View File

@@ -55,6 +55,9 @@ if __name__ == '__main__':
if args.mix and args.lang == "None": if args.mix and args.lang == "None":
print("When -mix is set, -lang must be set (support: ['zh', 'en'])") print("When -mix is set, -lang must be set (support: ['zh', 'en'])")
sys.exit(-1) sys.exit(-1)
elif args.mix and args.lang not in ['zh', 'en']:
print(f"language support: ['zh', 'en'] (invalid: {args.lang})")
sys.exit(-1)
# You can use your own checkpoint and tokenizer path. # You can use your own checkpoint and tokenizer path.
print('Loading model and tokenizer...') print('Loading model and tokenizer...')