Merge pull request #52 from hpcaitech/lzm_develop_2

fix parameter mistake
hpcaitech · May 10, 2022 · bdaa4d6 · bdaa4d6
2 parents 2dac783 + 81baea8
commit bdaa4d6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/energon/server/batch_manager.py b/energon/server/batch_manager.py
@@ -241,7 +241,7 @@ def processing_batch(self):
                 else:
                     input_ids = input_text
                 output = self.engine.run(input_ids)
-                self.pool.submit(self.publish_result, output, target_batch, start_time)
+                self.pool.submit(self.publish_result, output, target_batch)
                 # self.publish_result(output, target_batch, start_time)
                 # pub_thread = threading.Thread(target=self.publish_result, args=(output, target_batch, start_time))
                 # pub_thread.start()