Code

Add text embedding serving#214

Code

Merged

jonatanklosko merged 17 commits intoelixir-nx:mainelixir-nx/bumblebee:mainfrom coderrg:mainCopy head branch name to clipboardJun 2, 2023

+238-1

Commits

Commits on May 26, 2023

• Add text embedding serving

  coderrg

  committed
  5884438

  Copy full SHA for 5884438

Commits on May 27, 2023

• Set output_attribute default to :pooled_state

  coderrg

  committed
  6b2fe18

  Copy full SHA for 6b2fe18

• Add embedding_function option to allow for transforming output

  coderrg

  committed
  96b5c1a

  Copy full SHA for 96b5c1a

• Allow for output_attribute to be :none

  coderrg

  committed
  8e8ebfa

  Copy full SHA for 8e8ebfa

Commits on May 28, 2023

• Add support for embedding functions list and mean pooling function

  coderrg

  committed
  2caee5b

  Copy full SHA for 2caee5b

• Change :none option to nil for output_attribute

  coderrg

  committed
  8a27542

  Copy full SHA for 8a27542

Commits on Jun 1, 2023

• Move processing logic to embedding_fun

  coderrg

  committed
  4f2ddc5

  Copy full SHA for 4f2ddc5

• Update lib/bumblebee/text.ex

  Show description for bc42578
  

  coderrg

  and

  jonatanklosko

  authored
  bc42578

  Copy full SHA for bc42578

• Update lib/bumblebee/text.ex

  Show description for ac23753
  

  coderrg

  and

  jonatanklosko

  authored
  ac23753

  Copy full SHA for ac23753

Commits on Jun 2, 2023

• Update test/bumblebee/text/text_embedding_test.exs

  Show description for 652188d
  

  coderrg

  and

  jonatanklosko

  authored
  652188d

  Copy full SHA for 652188d

• Check if norm is 0 in normalize

  coderrg

  committed
  d8a5005

  Copy full SHA for d8a5005

• Allow for batched input in normalize

  coderrg

  committed
  8a83ed4

  Copy full SHA for 8a83ed4

• Update test/bumblebee/text/text_embedding_test.exs

  Show description for bb3ca9d
  

  coderrg

  and

  jonatanklosko

  authored
  bb3ca9d

  Copy full SHA for bb3ca9d

• Add test for normalized text embedding

  coderrg

  committed
  65a607d

  Copy full SHA for 65a607d

• Update lib/bumblebee/utils/nx.ex

  jonatanklosko

  authored
  6efca47

  Copy full SHA for 6efca47

• Update lib/bumblebee/utils/nx.ex

  jonatanklosko

  authored
  aacd6a1

  Copy full SHA for aacd6a1

• Update lib/bumblebee/utils/nx.ex

  jonatanklosko

  authored
  6e130f6

  Copy full SHA for 6e130f6