not-lain commited on
Commit
4a02b16
1 Parent(s): 0ae2d10

update model_type

Browse files
Files changed (2) hide show
  1. __init__.py +1 -1
  2. modeling_cerule_gemma.py +1 -1
__init__.py CHANGED
@@ -3,5 +3,5 @@ from .modeling_cerule_gemma import CeruleGemmaForCausalLM
3
 
4
  from transformers import AutoConfig, AutoModelForCausalLM
5
 
6
- AutoConfig.register("cerule-gemma", CeruleGemmaConfig)
7
  AutoModelForCausalLM.register(CeruleGemmaConfig, CeruleGemmaForCausalLM)
 
3
 
4
  from transformers import AutoConfig, AutoModelForCausalLM
5
 
6
+ AutoConfig.register("phi-msft", CeruleGemmaConfig)
7
  AutoModelForCausalLM.register(CeruleGemmaConfig, CeruleGemmaForCausalLM)
modeling_cerule_gemma.py CHANGED
@@ -2264,5 +2264,5 @@ class CeruleGemmaForCausalLM(GemmaForCausalLM, CeruleMetaForCausalLM):
2264
  return new_images
2265
 
2266
 
2267
- AutoConfig.register("cerule-gemma", CeruleGemmaConfig)
2268
  AutoModelForCausalLM.register(CeruleGemmaConfig, CeruleGemmaForCausalLM)
 
2264
  return new_images
2265
 
2266
 
2267
+ AutoConfig.register("phi-msft", CeruleGemmaConfig)
2268
  AutoModelForCausalLM.register(CeruleGemmaConfig, CeruleGemmaForCausalLM)