Skip to content

Commit de22157

Browse files
committed
Rebase and fix conflicts and type check
1 parent 0acdb78 commit de22157

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7928,11 +7928,13 @@ def prepare_tensors(self):
79287928
if len(experts) > 0:
79297929
raise ValueError(f"Unprocessed experts: {experts}")
79307930

7931+
79317932
@ModelBase.register("CogVLMForCausalLM")
79327933
class CogVLMVisionModel(MmprojModel):
7934+
79337935
def __init__(self, *args, **kwargs):
79347936
super().__init__(*args, **kwargs)
7935-
self.hparams_vision['num_attention_heads'] = self.hparams['num_heads']
7937+
self.hparams['num_attention_heads'] = self.hparams['num_heads']
79367938

79377939
def set_gguf_parameters(self):
79387940
super().set_gguf_parameters()
@@ -7956,6 +7958,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
79567958

79577959
return [(self.map_tensor_name(name), data_torch)]
79587960

7961+
79597962
@ModelBase.register("CogVLMForCausalLM")
79607963
class CogVLMModel(LlamaModel):
79617964
model_arch = gguf.MODEL_ARCH.COGVLM

0 commit comments

Comments
 (0)