Code

Onnx granite#2043

Code

Merged

xenova merged 9 commits intohuggingface:mainhuggingface/optimum:mainfrom gabe-l-hart:OnnxGranitegabe-l-hart/optimum:OnnxGraniteCopy head branch name to clipboardOct 31, 2024

+19-1

Commits

Commits on Oct 31, 2024

• feat(exporters/onnx): Add GraniteOnnxConfig and task support list

  Show description for bbf7d82
  gabe-l-hart

  committed
  bbf7d82

  Copy full SHA for bbf7d82

• feat: Add granite's normalized config for inference

  Show description for 63f66fc
  gabe-l-hart

  committed
  63f66fc

  Copy full SHA for 63f66fc

• feat(onnx opt): Add onnx optimization support for granite

  Show description for 767fe05
  gabe-l-hart

  committed
  767fe05

  Copy full SHA for 767fe05

• fix(onnx/granite): Use LlamaOnnxConfig as the base for GraniteOnnxConfig

  Show description for 1ab66df
  gabe-l-hart

  committed
  1ab66df

  Copy full SHA for 1ab66df

• fix(onnxruntime): Add "granite" to list of model types with grouped attention

  Show description for 4fc9ead
  gabe-l-hart

  committed
  4fc9ead

  Copy full SHA for 4fc9ead

• fix: Add granite to the list of models that require position_ids

  Show description for e6e5664
  gabe-l-hart

  committed
  e6e5664

  Copy full SHA for e6e5664

• fix(granite): Add MIN_TORCH_VERSION for recently fixed torch bug

  Show description for 551c47b
  gabe-l-hart

  committed
  551c47b

  Copy full SHA for 551c47b

• test(granite): Add tiny random granite test for onnx exporter

  Show description for 9529b51
  gabe-l-hart

  committed
  9529b51

  Copy full SHA for 9529b51

• tests(onnxruntime): Add granite to onnxruntime tests

  Show description for 52da69f
  gabe-l-hart

  committed
  52da69f

  Copy full SHA for 52da69f