ebe0ba91db
Signed-off-by: Kevin H. Luu <khluu000@gmail.com> Signed-off-by: khluu <khluu000@gmail.com> Co-authored-by: Kevin Luu <khluu@Kevins-MacBook-Pro.local>
26 lines
686 B
YAML
26 lines
686 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
|