Skip to content

Commit cee80d4

Browse files
committed
fix merge problem
1 parent 424807e commit cee80d4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llama-graph.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ void llm_graph_input_embd::set_input(const llama_ubatch * ubatch) {
4343
ggml_backend_tensor_set(tokens, ubatch->token, 0, n_tokens*ggml_element_size(tokens));
4444
}
4545

46-
if (ubatch->embd) {
46+
if (ubatch->embd && !ubatch->embd_tensor) {
4747
const int64_t n_embd = embd->ne[0];
4848
const int64_t n_tokens = ubatch->n_tokens;
4949

0 commit comments

Comments
 (0)