Enabling "Weight Loading Multiple GPU Test - Large Models" (#18020)

This commit is contained in:
Alexei-V-Ivanov-AMD 2025-05-12 15:05:33 -05:00 committed by GitHub
parent 289199feb6
commit e9c730c9bd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -703,6 +703,7 @@ steps:
- bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models.txt
- label: Weight Loading Multiple GPU Test - Large Models # optional
mirror_hardwares: [amdexperimental]
working_dir: "/vllm-workspace/tests"
num_gpus: 2
gpu: a100