Skip to content

Add initial support for TEI#40

Closed
ltowarek wants to merge 1 commit into
habana-mainfrom
ltowarek/tei
Closed

Add initial support for TEI#40
ltowarek wants to merge 1 commit into
habana-mainfrom
ltowarek/tei

Conversation

@ltowarek
Copy link
Copy Markdown

@ltowarek ltowarek commented Feb 14, 2024

Inspired by https://github.com/HabanaAI/optimum-habana-fork/tree/5dc75772f458842863d1b70d80b5c6265022dd1b/text-generation-inference

Main changes compared to the upstream:

  • Enabled python backend with support for HPU
  • Added Habana PT as a base image
  • Updated python requirements to match Habana PT image
    • Removed torch
    • Added transformers

text-embeddings-inference@0.6.0 can't be used as there are missing commits in candle.

Copy link
Copy Markdown

@kdamaszk kdamaszk left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

Comment thread text-embeddings-inference/README.md Outdated
Inspired by https://github.com/HabanaAI/optimum-habana-fork/tree/5dc75772f458842863d1b70d80b5c6265022dd1b/text-generation-inference

Main changes compared to the upstream:
* Enabled python backend with support for HPU
* Added Habana PT as a base image
* Updated python requirements to match Habana PT image
  * Removed torch
  * Added transformers

text-embeddings-inference@0.6.0 can't be used as there are missing commits in candle.
@ltowarek
Copy link
Copy Markdown
Author

ltowarek commented Mar 4, 2024

Moved to huggingface/tei-gaudi#1

@ltowarek ltowarek closed this Mar 4, 2024
@ltowarek ltowarek deleted the ltowarek/tei branch March 4, 2024 13:15
astachowiczhabana pushed a commit that referenced this pull request Nov 28, 2024
This reverts commit b06af09ef45bea2840403ab33e727f436db02b43.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants