Skip to content
This repository was archived by the owner on May 26, 2025. It is now read-only.

Add initial support for TEI on HPU#1

Merged
astachowiczhabana merged 1 commit into
huggingface:mainfrom
ltowarek:main
Mar 15, 2024
Merged

Add initial support for TEI on HPU#1
astachowiczhabana merged 1 commit into
huggingface:mainfrom
ltowarek:main

Conversation

@ltowarek
Copy link
Copy Markdown

@ltowarek ltowarek commented Mar 4, 2024

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

@astachowiczhabana
Copy link
Copy Markdown
Collaborator

astachowiczhabana commented Mar 4, 2024

LGTM, waiting for QA tests results.

Comment thread backends/python/server/requirements.txt
Comment thread README.md
@libinta
Copy link
Copy Markdown
Collaborator

libinta commented Mar 10, 2024 via email

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
Copy link
Copy Markdown
Collaborator

@astachowiczhabana astachowiczhabana left a comment

Choose a reason for hiding this comment

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

Change passes basic smoke tests

@astachowiczhabana astachowiczhabana merged commit 1782a2c into huggingface:main Mar 15, 2024
pi314ever pushed a commit to pi314ever/tei-gaudi that referenced this pull request Aug 20, 2024
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
pi314ever pushed a commit to pi314ever/tei-gaudi that referenced this pull request Aug 30, 2024
add python backend support for xlm-roberta type model
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants