jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

BaseInferencer batch_size and max_seq_len cast to int

Browse Source
This commit is contained in:
Francesco Bertolotti 2025-05-05 19:55:21 +02:00
parent 37cbaf8d92
commit 61b52844be
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
opencompass/openicl/icl_inferencer/icl_base_inferencer.py
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()