6521ccf286
Signed-off-by: Robert Shaw <robshaw@redhat.com> Signed-off-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com> Co-authored-by: Robert Shaw <robshaw@redhat.com>
26 lines
706 B
YAML
26 lines
706 B
YAML
group: Weight Loading
|
|
depends_on:
|
|
- image-build
|
|
steps:
|
|
- label: Weight Loading Multiple GPU # 33min
|
|
timeout_in_minutes: 45
|
|
working_dir: "/vllm-workspace/tests"
|
|
num_devices: 2
|
|
optional: true
|
|
source_file_dependencies:
|
|
- vllm/
|
|
- tests/weight_loading
|
|
commands:
|
|
- bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models.txt
|
|
|
|
# - label: Weight Loading Multiple GPU - Large Models # optional
|
|
# working_dir: "/vllm-workspace/tests"
|
|
# num_devices: 2
|
|
# device: a100
|
|
# optional: true
|
|
# source_file_dependencies:
|
|
# - vllm/
|
|
# - tests/weight_loading
|
|
# commands:
|
|
# - bash weight_loading/run_model_weight_loading_test.sh -c weight_loading/models-large.txt
|