Skip to content

Commit 6751e69

Browse files
committed
fix modern-bert class register
1 parent c6b84e2 commit 6751e69

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

convert_hf_to_gguf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3935,7 +3935,7 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
39353935
return super().modify_tensors(data_torch, name, bid)
39363936

39373937

3938-
@ModelBase.register("ModernBert", "ModernBertForMaskedLM", "ModernBertForSequenceClassification")
3938+
@ModelBase.register("ModernBertModel", "ModernBertForMaskedLM", "ModernBertForSequenceClassification")
39393939
class ModernBertModel(BertModel):
39403940
model_arch = gguf.MODEL_ARCH.MODERN_BERT
39413941

0 commit comments

Comments
 (0)