From 690fc31757a8dc10106e7f95fabe1397f21e0cc6 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Fri, 5 May 2023 15:28:08 +0200 Subject: [PATCH] fix(server): fix convert (#284) --- server/text_generation_server/utils/convert.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/server/text_generation_server/utils/convert.py b/server/text_generation_server/utils/convert.py index 8eddd76f..1bdedf58 100644 --- a/server/text_generation_server/utils/convert.py +++ b/server/text_generation_server/utils/convert.py @@ -81,7 +81,9 @@ def convert_files(pt_files: List[Path], st_files: List[Path]): N = len(pt_files) # We do this instead of using tqdm because we want to parse the logs with the launcher - start = datetime.datetime.now() + for i, (pt_file, sf_file) in enumerate(zip(pt_files, st_files)): + start = datetime.datetime.now() + convert_file(pt_file, sf_file) elapsed = datetime.datetime.now() - start logger.info(f"Convert: [{i + 1}/{N}] -- Took: {elapsed}")