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:
414414 config :
415415 - backend : " bitsandbytes"
416416 test_location : " bnb"
417+ additional_deps : ["peft"]
417418 - backend : " gguf"
418419 test_location : " gguf"
420+ additional_deps : []
419421 - backend : " torchao"
420422 test_location : " torchao"
423+ additional_deps : []
421424 - backend : " optimum_quanto"
422425 test_location : " quanto"
426+ additional_deps : []
423427 runs-on :
424428 group : aws-g6e-xlarge-plus
425429 container :
@@ -437,8 +441,8 @@ jobs:
437441 python -m venv /opt/venv && export PATH="/opt/venv/bin:$PATH"
438442 python -m uv pip install -e [quality,test]
439443 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, ' ') }}
442446 fi
443447 python -m uv pip install pytest-reportlog
444448 - name : Environment
You can’t perform that action at this time.
0 commit comments