This commit is contained in:
gauthiier 2022-02-09 18:18:41 +01:00
parent 04efe2cc4f
commit b77d9122ef

View File

@ -99,7 +99,7 @@ def main() -> int:
if output_dir.is_dir(): if output_dir.is_dir():
exts = ['.json', '.gz', '.bin'] exts = ['.json', '.gz', '.bin']
files = [x for x in output_dir.glob('*') if x.suffix in exts] files = [x for x in output_dir.glob('*') if x.suffix in exts]
if len(files) == 3: if len(files) == 4:
print("Token + model already exists > " + output_dir.name) print("Token + model already exists > " + output_dir.name)
q = input("Continue? [y/n]") q = input("Continue? [y/n]")
if q != 'y': if q != 'y':