@@ -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 ] ] ,
0 commit comments