Skip to content

Commit 820cee1

Browse files
committed
revert incorrect change
1 parent 5821d6c commit 820cee1

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

src/llama-graph.cpp

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1602,9 +1602,7 @@ void llm_graph_context::build_pooling(
16021602
cur = ggml_add(ctx0, cur, cls_out_b);
16031603
}
16041604
} else {
1605-
// Some models may not have either classification heads
1606-
// In this case, just use the CLS/pooled embedding directly
1607-
cur = inp;
1605+
GGML_ABORT("RANK pooling requires either cls+cls_b or cls_out+cls_out_b");
16081606
}
16091607
} break;
16101608
default:

0 commit comments

Comments
 (0)