Commit Graph

4 Commits

Author SHA1 Message Date
David Corvoysier
1e4e406d77 test(neuron): avoid using image sha when exporting models
We now manually evaluate the apparent hash of the neuron backend by
combining the hash of the neuron backend directory and Dockerfile.
This new hash is used to identify exported neuron models instead of the
image sha.
This has two benefits:
- it changes less frequently (only hwen the neuron backend changes),
  which means less neuron models being pushed to the hub,
- it can be evaluated locally, meaning that running the tests once
  locally will export the models before the CI uses them.
2025-02-21 15:32:13 +00:00
David Corvoysier
7d6ff64c13
test(neuron): use smaller llama model 2025-02-20 16:14:42 +01:00
David Corvoysier
49dfdc3f8a
fix(neuron): export models from container in test fixtures
The neuron tests require models to have been previously exported and
cached on the hub. This is done automatically by the neuron.model
fixture the first time the tests are ran for a specific version.
This fixture used to export the models using optimum-neuron directly,
but this package is not necessarily present on the system.
Instead, it is now done through the neuron TGI itself, since it
contains all the tools required to export the models.
Note that since the CI runs docker in docker (dind) it does not seem
possible to share a volume between the CI container and the container
used to export the model.
For that reason, a specific image with a modified entrypoint is built
on-the-fly when a model export is required.
2025-02-20 16:14:42 +01:00
David Corvoysier
2c37e8acbe
test(neuron): merge integration tests and fixtures 2025-02-20 16:14:41 +01:00