Skip to content

Commit 64cd4b1

Browse files
author
Joan Martinez
committed
fix: fix linting and editor
1 parent c1c0f4d commit 64cd4b1

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

convert-hf-to-gguf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2714,6 +2714,7 @@ def write_tensors(self):
27142714

27152715
self.gguf_writer.add_tensor(new_name, data)
27162716

2717+
27172718
@Model.register("JinaBertModel")
27182719
class JinaBertModel(BertModel):
27192720
model_arch = gguf.MODEL_ARCH.JINA_BERT

llama.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5124,7 +5124,7 @@ static bool llm_load_tensors(
51245124

51255125
auto & layer = model.layers[i]; // JinaBertLayer
51265126

5127-
layer.wq = ml.create_tensor(ctx_split, tn(LLM_TENSOR_ATTN_Q, "weight", i), {n_embd, n_embd});
5127+
layer.wq = ml.create_tensor(ctx_split, tn(LLM_TENSOR_ATTN_Q, "weight", i), {n_embd, n_embd});
51285128
layer.bq = ml.create_tensor(ctx_layer, tn(LLM_TENSOR_ATTN_Q, "bias", i), {n_embd});
51295129

51305130
layer.wk = ml.create_tensor(ctx_split, tn(LLM_TENSOR_ATTN_K, "weight", i), {n_embd, n_embd_gqa});

0 commit comments

Comments
 (0)