File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -92,7 +92,6 @@ def get_tensors(self) -> Iterator[tuple[str, Tensor]]:
92
92
93
93
def set_gguf_parameters (self ):
94
94
self .gguf_writer .add_name (self .dir_model .name )
95
- print (f'self.block_count { self .block_count } ' )
96
95
self .gguf_writer .add_block_count (self .block_count )
97
96
98
97
if (n_ctx := self .find_hparam (["max_position_embeddings" , "n_ctx" ], optional = True )) is not None :
@@ -138,7 +137,6 @@ def set_gguf_parameters(self):
138
137
def write_tensors (self ):
139
138
block_count = self .hparams .get ("n_layers" , self .hparams .get ("num_hidden_layers" , self .hparams .get ("n_layer" )))
140
139
tensor_map = gguf .get_tensor_name_map (self .model_arch , block_count )
141
- print (f'Block_count { block_count } with tensor_map { tensor_map } ' )
142
140
for name , data_torch in self .get_tensors ():
143
141
# we don't need these
144
142
if name .endswith ((".attention.masked_bias" , ".attention.bias" , ".attention.rotary_emb.inv_freq" )):
@@ -2188,6 +2186,9 @@ def get_tensors(self):
2188
2186
yield name , data
2189
2187
2190
2188
2189
+ JinaBertForMaskedML = JinaBertModel
2190
+
2191
+
2191
2192
@Model .register ("GemmaForCausalLM" )
2192
2193
class GemmaModel (Model ):
2193
2194
model_arch = gguf .MODEL_ARCH .GEMMA
Original file line number Diff line number Diff line change @@ -12476,8 +12476,9 @@ static void ggml_compute_forward_alibi_f32(
12476
12476
12477
12477
for (int64_t i = 0; i < ne0; i++) {
12478
12478
for (int64_t j = 0; j < ne1; j++) {
12479
+ float * const src = (float *)((char *) src0->data + i*nb0 + j*nb1 + k*nb2);
12479
12480
float * pdst = (float *)((char *) dst->data + i*nb0 + j*nb1 + k*nb2);
12480
- pdst[0] = -1.0f * i * m_k;
12481
+ pdst[0] = i * m_k + src[0] ;
12481
12482
}
12482
12483
}
12483
12484
}
You can’t perform that action at this time.
0 commit comments