Whisper: pin `transformers` version (#1296)

This commit is contained in:
Jambay Kinley 2024-08-12 08:39:42 -07:00 коммит произвёл GitHub
Родитель 1f8f4f8249
Коммит b1e051e061
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
2 изменённых файлов: 6 добавлений и 2 удалений

Просмотреть файл

@ -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__":