Add support for nomic_bert

Prevents unsupported model warnings
huggingface · Feb 15, 2024 · 2fa6218 · 2fa6218
1 parent 3d1e0cc
commit 2fa6218
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/models.js b/src/models.js
@@ -1471,6 +1471,12 @@ export class BertForQuestionAnswering extends BertPreTrainedModel {
 }
 //////////////////////////////////////////////////
 
+//////////////////////////////////////////////////
+// NomicBert models
+export class NomicBertPreTrainedModel extends PreTrainedModel { }
+export class NomicBertModel extends NomicBertPreTrainedModel { }
+//////////////////////////////////////////////////
+
 //////////////////////////////////////////////////
 // RoFormer models
 export class RoFormerPreTrainedModel extends PreTrainedModel { }
@@ -5177,6 +5183,7 @@ export class PretrainedMixin {
 
 const MODEL_MAPPING_NAMES_ENCODER_ONLY = new Map([
     ['bert', ['BertModel', BertModel]],
+    ['nomic_bert', ['NomicBertModel', NomicBertModel]],
     ['roformer', ['RoFormerModel', RoFormerModel]],
     ['electra', ['ElectraModel', ElectraModel]],
     ['esm', ['EsmModel', EsmModel]],