Code

ROCm support#293

Code

Closed

fxmarty wants to merge 10 commits intohuggingface:mainhuggingface/text-embeddings-inference:mainfrom fxmarty:rocm-supportfxmarty/text-embeddings-inference:rocm-supportCopy head branch name to clipboard

+761-80

Commits

Commits on Jun 17, 2024

• add dockerfile

  fxmarty

  committed
  2fc644c

  Copy full SHA for 2fc644c

Commits on Jun 18, 2024

• working cls pooling

  fxmarty

  committed
  37d2931

  Copy full SHA for 37d2931

• add layers

  fxmarty

  committed
  8584b6d

  Copy full SHA for 8584b6d

• Merge branch 'main' into rocm-support

  fxmarty

  committed
  6cdd454

  Copy full SHA for 6cdd454

Commits on Jun 19, 2024

• support mean pooling in python backend

  fxmarty

  committed
  2a2993a

  Copy full SHA for 2a2993a

• fix dockerfile and install

  fxmarty

  committed
  36b3a72

  Copy full SHA for 36b3a72

• add tests

  fxmarty

  committed
  a8c02db

  Copy full SHA for a8c02db

• tests instructions

  fxmarty

  committed
  35cc5b8

  Copy full SHA for 35cc5b8

• add rocm image builder

  fxmarty

  committed
  309d255

  Copy full SHA for 309d255

• add reference tensors

  fxmarty

  committed
  ae3da10

  Copy full SHA for ae3da10