wq
Browse files- modelConfig.py +1 -1
modelConfig.py
CHANGED
|
@@ -5,7 +5,7 @@ class OBIConfig(PretrainedConfig):
|
|
| 5 |
def __init__(self,
|
| 6 |
model_type="OBILanguageModel",
|
| 7 |
auto_map={
|
| 8 |
-
|
| 9 |
"AutoModel": "modelLM.OBILanguageModel",
|
| 10 |
"AutoModelForCausalLM": "modelLM.OBILanguageModel",
|
| 11 |
"AutoModelForQuestionAnswering": "modelLM.OBILanguageModel"
|
|
|
|
| 5 |
def __init__(self,
|
| 6 |
model_type="OBILanguageModel",
|
| 7 |
auto_map={
|
| 8 |
+
"AutoConfig": "modelConfig.OBIConfig",
|
| 9 |
"AutoModel": "modelLM.OBILanguageModel",
|
| 10 |
"AutoModelForCausalLM": "modelLM.OBILanguageModel",
|
| 11 |
"AutoModelForQuestionAnswering": "modelLM.OBILanguageModel"
|