From 11fc50be9824477027b21a2702ccce6b0c56cdb6 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 07 六月 2023 20:01:02 +0800
Subject: [PATCH] Update infer.py batch_size_token (#609)
---
funasr/fileio/datadir_writer.py | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/funasr/fileio/datadir_writer.py b/funasr/fileio/datadir_writer.py
index bafdf98..67ec61c 100644
--- a/funasr/fileio/datadir_writer.py
+++ b/funasr/fileio/datadir_writer.py
@@ -57,6 +57,7 @@
self.keys.add(key)
self.fd.write(f"{key} {value}\n")
+ self.fd.flush()
def __exit__(self, exc_type, exc_val, exc_tb):
self.close()
--
Gitblit v1.9.1