fix loading issue
Browse files- config.json +0 -1
- configuration.py +0 -2
config.json
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
{
|
| 2 |
-
"model_type": "user",
|
| 3 |
"transformers_version": "4.20.0"
|
| 4 |
}
|
|
|
|
| 1 |
{
|
|
|
|
| 2 |
"transformers_version": "4.20.0"
|
| 3 |
}
|
configuration.py
CHANGED
|
@@ -2,7 +2,5 @@ from transformers import PretrainedConfig
|
|
| 2 |
|
| 3 |
|
| 4 |
class UserModelConfig(PretrainedConfig):
|
| 5 |
-
model_type = "user"
|
| 6 |
-
|
| 7 |
def __init__(self, **kwargs):
|
| 8 |
super().__init__(**kwargs)
|
|
|
|
| 2 |
|
| 3 |
|
| 4 |
class UserModelConfig(PretrainedConfig):
|
|
|
|
|
|
|
| 5 |
def __init__(self, **kwargs):
|
| 6 |
super().__init__(**kwargs)
|