Update modeling_business.py
Browse files- modeling_business.py +1 -1
modeling_business.py
CHANGED
|
@@ -23,7 +23,7 @@ class BusinessClassifierConfig(PretrainedConfig):
|
|
| 23 |
|
| 24 |
|
| 25 |
class BusinessClassifier(PreTrainedModel):
|
| 26 |
-
config_class =
|
| 27 |
|
| 28 |
def __init__(self, config):
|
| 29 |
super().__init__(config)
|
|
|
|
| 23 |
|
| 24 |
|
| 25 |
class BusinessClassifier(PreTrainedModel):
|
| 26 |
+
config_class = BusinessClassifierConfig
|
| 27 |
|
| 28 |
def __init__(self, config):
|
| 29 |
super().__init__(config)
|