This commit is contained in:
Francesco Bertolotti 2025-05-29 14:24:05 +08:00 committed by GitHub
commit 826aae6869
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -46,8 +46,8 @@ class BaseInferencer:
self.model = model
self.max_seq_len = max_seq_len
self.batch_size = batch_size
self.max_seq_len = int(max_seq_len)
self.batch_size = int(batch_size)
self.output_json_filepath = output_json_filepath
self.output_json_filename = output_json_filename
self.is_main_process = is_main_process()