text-embeddings-inference icon indicating copy to clipboard operation
text-embeddings-inference copied to clipboard

Add integration tests for Gaudi

Open baptistecolle opened this issue 6 months ago • 2 comments

What does this PR do?

This PR adds an integration test for Gaudi, with the goal of eventually including it in the CI pipeline. The CI pipeline will be a scheduled nightly test to check that there is no regression in the Gaudi TEI image.

Unfortunately, while working on this, I discovered that TEI is not compatible with Gaudi 1. As a result, it’s currently not possible to enable CI for this test, since we’re still in the process of setting up a CI cluster for Gaudi 2.

This integration test starts a TEI image on HPU, embeds a sample text, and verifies that the resulting embedding matches the expected output.

baptistecolle avatar Apr 22 '25 13:04 baptistecolle