Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Link to nix owned lib

This commit is contained in:
Nicolas Patry 2024-09-16 10:56:32 +02:00
parent 83ee55a617
commit ec6fe324c6
No known key found for this signature in database
GPG Key ID: 64AF4752B2967863
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/nix_integration_tests.yaml vendored
View File

@ -32,7 +32,10 @@ jobs:
run: nix develop .#test --command echo "Ok"
- name: Python tests.
run: |
export LD_LIBRARY_PATH=$(nix-build -E 'with import <nixpkgs> { config = { allowUnfree = true; }; }; linuxPackages.nvidia_x11')
sudo mkdir -p /run/opengl-driver/lib/
export NVIDIA=$(nix-build -E 'with import <nixpkgs> { config = { allowUnfree = true; }; }; linuxPackages.nvidia_x11')
sudo find $NVIDIA . -name 'libcuda.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
sudo find $NVIDIA . -name 'libnvidia-ml.so*' -exec ln -s {} /run/opengl-driver/lib/ \;
nix develop .#test --command pytest -svv integration-tests/ --release
env:
HF_TOKEN: ${{ secrets.HF_TOKEN }}