Skip to content

Commit eb7dd02

Browse files
authored
Fix Ernie 4.5 naming (#1473)
1 parent a6b8659 commit eb7dd02

File tree

2 files changed

+5
-8
lines changed

2 files changed

+5
-8
lines changed

src/models.js

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6950,11 +6950,11 @@ export class MistralForCausalLM extends MistralPreTrainedModel { }
69506950

69516951
//////////////////////////////////////////////////
69526952
// ERNIE-4.5 models
6953-
export class Ernie4_5_PretrainedModel extends PreTrainedModel { }
6953+
export class Ernie4_5PreTrainedModel extends PreTrainedModel { }
69546954

6955-
export class Ernie4_5_Model extends Ernie4_5_PretrainedModel { }
6955+
export class Ernie4_5Model extends Ernie4_5PreTrainedModel { }
69566956

6957-
export class Ernie4_5_ForCausalLM extends Ernie4_5_PretrainedModel { }
6957+
export class Ernie4_5ForCausalLM extends Ernie4_5PreTrainedModel { }
69586958
//////////////////////////////////////////////////
69596959

69606960

@@ -8041,7 +8041,7 @@ const MODEL_MAPPING_NAMES_DECODER_ONLY = new Map([
80418041
['mpt', ['MptModel', MptModel]],
80428042
['opt', ['OPTModel', OPTModel]],
80438043
['mistral', ['MistralModel', MistralModel]],
8044-
['ernie4_5', ['Ernie4_5_Model', Ernie4_5_Model]],
8044+
['ernie4_5', ['Ernie4_5Model', Ernie4_5Model]],
80458045
['starcoder2', ['Starcoder2Model', Starcoder2Model]],
80468046
['falcon', ['FalconModel', FalconModel]],
80478047
['stablelm', ['StableLmModel', StableLmModel]],
@@ -8155,7 +8155,7 @@ const MODEL_FOR_CAUSAL_LM_MAPPING_NAMES = new Map([
81558155
['opt', ['OPTForCausalLM', OPTForCausalLM]],
81568156
['mbart', ['MBartForCausalLM', MBartForCausalLM]],
81578157
['mistral', ['MistralForCausalLM', MistralForCausalLM]],
8158-
['ernie4_5', ['Ernie4_5_ForCausalLM', Ernie4_5_ForCausalLM]],
8158+
['ernie4_5', ['Ernie4_5ForCausalLM', Ernie4_5ForCausalLM]],
81598159
['starcoder2', ['Starcoder2ForCausalLM', Starcoder2ForCausalLM]],
81608160
['falcon', ['FalconForCausalLM', FalconForCausalLM]],
81618161
['trocr', ['TrOCRForCausalLM', TrOCRForCausalLM]],

src/tokenizers.js

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4352,8 +4352,6 @@ export class CohereTokenizer extends PreTrainedTokenizer { }
43524352

43534353
export class MgpstrTokenizer extends PreTrainedTokenizer { }
43544354

4355-
export class Ernie4_5_Tokenizer extends PreTrainedTokenizer { }
4356-
43574355
/**
43584356
* Helper class which is used to instantiate pretrained tokenizers with the `from_pretrained` function.
43594357
* The chosen tokenizer class is determined by the type specified in the tokenizer config.
@@ -4408,7 +4406,6 @@ export class AutoTokenizer {
44084406
Grok1Tokenizer,
44094407
CohereTokenizer,
44104408
MgpstrTokenizer,
4411-
Ernie4_5_Tokenizer,
44124409

44134410
// Base case:
44144411
PreTrainedTokenizer,

0 commit comments

Comments
 (0)