mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-21 14:52:20 +00:00
Using the cache on both jobs.
This commit is contained in:
parent
110b9a0b4c
commit
911f82a34b
2
.github/workflows/build.yaml
vendored
2
.github/workflows/build.yaml
vendored
@ -183,7 +183,7 @@ jobs:
|
||||
make install-integration-tests
|
||||
- name: Run tests
|
||||
run: |
|
||||
export DOCKER_VOLUME=/mnt/cache
|
||||
export DOCKER_VOLUME=/mnt/hf_cache
|
||||
export DOCKER_IMAGE=${{ needs.build-and-push.outputs.docker_image }}
|
||||
export DOCKER_DEVICES=${{ needs.build-and-push.outputs.docker_devices }}
|
||||
export HF_TOKEN=${{ secrets.HF_TOKEN }}
|
||||
|
2
.github/workflows/nix_integration_tests.yaml
vendored
2
.github/workflows/nix_integration_tests.yaml
vendored
@ -37,7 +37,7 @@ jobs:
|
||||
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/ \;
|
||||
sudo chown -R runner /mnt/hf_cache/hub
|
||||
nix develop .#test --command pytest -svvx integration-tests/ --release
|
||||
nix develop .#test --command pytest -svv integration-tests/ --release
|
||||
env:
|
||||
HF_TOKEN: ${{ secrets.HF_TOKEN }}
|
||||
HF_HOME: /mnt/hf_cache/
|
||||
|
Loading…
Reference in New Issue
Block a user