File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -414,12 +414,16 @@ jobs:
414
414
config :
415
415
- backend : " bitsandbytes"
416
416
test_location : " bnb"
417
+ additional_deps : ["peft"]
417
418
- backend : " gguf"
418
419
test_location : " gguf"
420
+ additional_deps : []
419
421
- backend : " torchao"
420
422
test_location : " torchao"
423
+ additional_deps : []
421
424
- backend : " optimum_quanto"
422
425
test_location : " quanto"
426
+ additional_deps : []
423
427
runs-on :
424
428
group : aws-g6e-xlarge-plus
425
429
container :
@@ -437,8 +441,8 @@ jobs:
437
441
python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
438
442
python -m uv pip install -e [quality,test]
439
443
python -m uv pip install -U ${{ matrix.config.backend }}
440
- if [ "${{ matrix.config.backend }}" = "bitsandbytes " ]; then
441
- python -m uv pip install peft
444
+ if [ "${{ join( matrix.config.additional_deps, ' ') }}" ! = "" ]; then
445
+ python -m uv pip install ${{ join(matrix.config.additional_deps, ' ') }}
442
446
fi
443
447
python -m uv pip install pytest-reportlog
444
448
- name : Environment
You can’t perform that action at this time.
0 commit comments