diff --git a/tests/sparseml/transformers/obcq/obcq_configs/completion/gpu/llama_7b_quant.yaml b/tests/sparseml/transformers/obcq/obcq_configs/completion/gpu/llama_7b_quant.yaml index 88a9535e7d9..8085e3e0a35 100644 --- a/tests/sparseml/transformers/obcq/obcq_configs/completion/gpu/llama_7b_quant.yaml +++ b/tests/sparseml/transformers/obcq/obcq_configs/completion/gpu/llama_7b_quant.yaml @@ -1,4 +1,4 @@ -cadence: "nightly" +cadence: "weekly" test_type: "regression" model: "zoo:llama2-7b-llama2_pretrain-base" dataset: open_platypus diff --git a/tests/sparseml/transformers/obcq/obcq_configs/completion/gpu/llama_7b_quant_and_sparse.yaml b/tests/sparseml/transformers/obcq/obcq_configs/completion/gpu/llama_7b_quant_and_sparse.yaml index 3f430569b9f..50cb1bc286f 100644 --- a/tests/sparseml/transformers/obcq/obcq_configs/completion/gpu/llama_7b_quant_and_sparse.yaml +++ b/tests/sparseml/transformers/obcq/obcq_configs/completion/gpu/llama_7b_quant_and_sparse.yaml @@ -1,4 +1,4 @@ -cadence: "nightly" +cadence: "weekly" test_type: "regression" model: "zoo:llama2-7b-llama2_pretrain-base" dataset: open_platypus diff --git a/tests/sparseml/transformers/obcq/obcq_configs/sparse/gpu/llama_7b_sparse.yaml b/tests/sparseml/transformers/obcq/obcq_configs/sparse/gpu/llama_7b_sparse.yaml index 1316258fbda..c58062707db 100644 --- a/tests/sparseml/transformers/obcq/obcq_configs/sparse/gpu/llama_7b_sparse.yaml +++ b/tests/sparseml/transformers/obcq/obcq_configs/sparse/gpu/llama_7b_sparse.yaml @@ -4,4 +4,4 @@ model: "zoo:llama2-7b-llama2_pretrain-base" dataset: open_platypus recipe: "tests/sparseml/transformers/obcq/recipes/sparse.yaml" sparsity: 0.3 -device: "cuda:0" \ No newline at end of file +device: "cuda:1" \ No newline at end of file