Whisper: pin `transformers` version (#1296)
This commit is contained in:
Родитель
1f8f4f8249
Коммит
b1e051e061
|
@ -2,4 +2,6 @@ neural-compressor
|
|||
onnxruntime_extensions>=0.9.0
|
||||
tabulate
|
||||
torch>=1.13.1
|
||||
transformers>=4.23.1
|
||||
# exported model is incompatible after transformers 4.43.0
|
||||
# produces empty text
|
||||
transformers>=4.23.1,<4.43.0
|
||||
|
|
|
@ -118,8 +118,10 @@ def main(raw_args=None):
|
|||
# get output
|
||||
input_data, _ = dataset[0]
|
||||
input_data = OnnxEvaluator.format_input(input_data, olive_model.io_config)
|
||||
# output is an list of numpy arrays, first element is the transcription 1Xnum_return_sequences
|
||||
# [["transcription1", "transcription2"]]
|
||||
output = olive_model.run_session(session, input_data)
|
||||
return output[0][0]
|
||||
return output[0][0][0]
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
|
Загрузка…
Ссылка в новой задаче