Skip to content

Commit 634c3ca

Browse files
committed
add exception in test-llama-archs
1 parent c6aa598 commit 634c3ca

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/test-llama-archs.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -391,7 +391,7 @@ static bool arch_supported(const llm_arch arch) {
391391
if (arch == LLM_ARCH_WAVTOKENIZER_DEC) {
392392
return false; // FIXME CUDA backend crashes.
393393
}
394-
if (arch == LLM_ARCH_GEMMA4) {
394+
if (arch == LLM_ARCH_GEMMA4 || arch == LLM_ARCH_GEMMA4_ASSISTANT) {
395395
return false; // FIXME @ngxson
396396
}
397397
if (arch == LLM_ARCH_LLAMA_EMBED || arch == LLM_ARCH_GEMMA_EMBEDDING || arch == LLM_ARCH_T5ENCODER) {
@@ -446,7 +446,7 @@ static int save_models(const llm_arch target_arch, const size_t seed, const ggml
446446
if (target_arch != LLM_ARCH_UNKNOWN && arch != target_arch) {
447447
continue;
448448
}
449-
if (arch == LLM_ARCH_GEMMA4) {
449+
if (arch == LLM_ARCH_GEMMA4 || arch == LLM_ARCH_GEMMA4_ASSISTANT) {
450450
continue; // FIXME: ISWA KV cache initialization needs more fixture params
451451
}
452452
for (bool moe : {false, true}) {
@@ -549,7 +549,7 @@ static int test_backends(const llm_arch target_arch, const size_t seed, const gg
549549
if (target_arch != LLM_ARCH_UNKNOWN && arch != target_arch) {
550550
continue;
551551
}
552-
if (arch == LLM_ARCH_GEMMA4) {
552+
if (arch == LLM_ARCH_GEMMA4 || arch == LLM_ARCH_GEMMA4_ASSISTANT) {
553553
continue; // FIXME: ISWA KV cache initialization needs more fixture params
554554
}
555555

0 commit comments

Comments
 (0)