Code

Align I/O with Inference API#99

Code

Merged

alvarobartt merged 26 commits intomainhuggingface/huggingface-inference-toolkit:mainfrom inference-api-alignmenthuggingface/huggingface-inference-toolkit:inference-api-alignmentCopy head branch name to clipboardDec 12, 2024

+277-149

Commits

Commits on Nov 12, 2024

• Fix task type-hint and remove extra space in logging

  alvarobartt

  committed
  d20528a

• Align transformers and diffusers inputs with Inference API

  alvarobartt

  committed
  be146c8

• Remove duplicated sentencepiece extra requirement

  alvarobartt

  committed
  0b61436

Commits on Nov 13, 2024

• Remove pipeline.task check for sentence-transformers

  alvarobartt

  committed
  49254e9

Commits on Nov 15, 2024

• Add warning and pop unsupported parameters

  alvarobartt

  committed
  b45c40a

• Fix sentence-transformers pipeline type-hints

  alvarobartt

  committed
  b9dec32

• Update sentence-ranking type-hints

  alvarobartt

  committed
  77c2bb2

• Add missing type-hints and clear code a bit

  alvarobartt

  committed
  9b4fc67

• Fix failing sentence-transformers tests due to input parsing

  alvarobartt

  committed
  c1d519a

• Fix "table-question-answering" payload check

  alvarobartt

  committed
  7f0d84d

• Fix "zero-shot-classification" payload check

  alvarobartt

  committed
  307b27f

• Check that payload is dict in advance

  alvarobartt

  committed
  d3d2b5e

• Fix HuggingFaceHandler errors and checks

  alvarobartt

  committed
  64cbeb1

• Fix sentence-transformers pipelines as those don't have parameters

  alvarobartt

  committed
  8cbd4be

• Fix INPUT to input_data fixture

  alvarobartt

  committed
  0053e97

• Fix quality in tests/unit/test_handler.py

  alvarobartt

  committed
  b9dbf58

Commits on Nov 18, 2024

• Make parameters default to empty dict instead of None

  alvarobartt

  committed
  d764e44

Commits on Dec 2, 2024

• Add note on token-classification / ner task

  Show description for 5fbe5af
  alvarobartt

  committed
  5fbe5af

• Update version in setup.py

  alvarobartt

  committed
  21ab873

Commits on Dec 3, 2024

• Fix generate_kwargs payload handling for text2text-based tasks

  alvarobartt

  committed
  7a225e2

• Fix generate_kwargs handling to move to flatten first-level dict

  Show description for cd9ebe7
  

  alvarobartt

  and

  hanouticelina

  committed
  cd9ebe7

• Update generate_kwargs handling as sometimes required

  alvarobartt

  committed
  280101d

Commits on Dec 4, 2024

• Remove generate from supported generation kwargs key names

  alvarobartt

  committed
  42cd852

• Update SentenceRankingPipeline to handle query-texts pipelines

  Show description for 01cd7a8
  alvarobartt

  committed
  01cd7a8

• Update typing and fix sentence-transformers tests

  alvarobartt

  committed
  4ffcdfd

Commits on Dec 12, 2024

• Upgrade transformers, sentence-transformers and peft dependencies

  alvarobartt

  committed
  9d87331