Как экспортировать настроенную модель SetFit в формат ONNX? ⇐ Python
-
Anonymous
Как экспортировать настроенную модель SetFit в формат ONNX?
SetFit имеет встроенную функциональность для экспорта в ONNX, но нет реальной документации о том, как это сделать. Что следует передать для параметров setfit_model и inputs в функции export_onnx_setfit_model?
# Загрузить локально сохраненную предварительно обученную модель setfit_model = SetFitModel.from_pretrained("путь/к/моей/модели", local_files_only=True) # Создаем оболочку SetFit onnx onnx_setfit_model = OnnxSetFitModel(model_body=setfit_model.model_body) # Экспорт модели Export_onnx_setfit_model(setfit_model=onnx_setfit_model, inputs=???, output_path="путь/к/выходу")
SetFit имеет встроенную функциональность для экспорта в ONNX, но нет реальной документации о том, как это сделать. Что следует передать для параметров setfit_model и inputs в функции export_onnx_setfit_model?
# Загрузить локально сохраненную предварительно обученную модель setfit_model = SetFitModel.from_pretrained("путь/к/моей/модели", local_files_only=True) # Создаем оболочку SetFit onnx onnx_setfit_model = OnnxSetFitModel(model_body=setfit_model.model_body) # Экспорт модели Export_onnx_setfit_model(setfit_model=onnx_setfit_model, inputs=???, output_path="путь/к/выходу")
