update model_type
Browse files- __init__.py +1 -1
- 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("
|
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("
|
2268 |
AutoModelForCausalLM.register(CeruleGemmaConfig, CeruleGemmaForCausalLM)
|
|
|
2264 |
return new_images
|
2265 |
|
2266 |
|
2267 |
+
AutoConfig.register("phi-msft", CeruleGemmaConfig)
|
2268 |
AutoModelForCausalLM.register(CeruleGemmaConfig, CeruleGemmaForCausalLM)
|