fix(server): fix convert (huggingface#284)

c3-pakazad · May 5, 2023 · 690fc31 · 690fc31
1 parent e68509a
commit 690fc31
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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}")