Skip to content

Commit 4e58ca4

Browse files
committed
test-model-random : avoid testing too many sequences for now
It makes the test longer for little benefit.
1 parent 18d2055 commit 4e58ca4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/test-model-random.cpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -818,6 +818,8 @@ struct model_variant {
818818
case LLM_ARCH_DOTS1:
819819
case LLM_ARCH_ARCEE:
820820
case LLM_ARCH_ERNIE4_5:
821+
case LLM_ARCH_HUNYUAN_MOE:
822+
case LLM_ARCH_SMOLLM3:
821823
case LLM_ARCH_UNKNOWN:
822824
break;
823825
}
@@ -1093,7 +1095,7 @@ int main(int argc, char ** argv) {
10931095
// const auto n_vocab = llama_vocab_n_tokens(llama_model_get_vocab(model));
10941096
// const auto n_embd = llama_model_n_embd(model);
10951097

1096-
for (int32_t n_seq_max : { 1, 2, 5, 13 } ) {
1098+
for (int32_t n_seq_max : { 1, 2, 5 }) {
10971099

10981100
// TODO(later): context shift testing
10991101
for (int32_t n_ctx : { n_seq_len * n_seq_max }) {

0 commit comments

Comments
 (0)