Skip to content

Commit a21e755

Browse files
committed
fix NomicBertModel
1 parent 1a0485d commit a21e755

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

convert_hf_to_gguf.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3507,6 +3507,8 @@ def modify_tensors(self, data_torch: Tensor, name: str, bid: int | None) -> Iter
35073507

35083508
@ModelBase.register("NomicBertModel")
35093509
class NomicBertModel(BertModel):
3510+
model_arch = gguf.MODEL_ARCH.BERT
3511+
35103512
def __init__(self, dir_model: Path, ftype: gguf.LlamaFileType, fname_out: Path, **kwargs: Any):
35113513
hparams = kwargs.pop("hparams", None)
35123514
if hparams is None:

0 commit comments

Comments
 (0)