Upload model
Browse files- modeling_basnet.py +2 -0
modeling_basnet.py
CHANGED
|
@@ -148,6 +148,8 @@ class BasicBlock(nn.Module):
|
|
| 148 |
|
| 149 |
|
| 150 |
class BASNetModel(PreTrainedModel):
|
|
|
|
|
|
|
| 151 |
def __init__(self, config: BASNetConfig) -> None:
|
| 152 |
super().__init__(config)
|
| 153 |
|
|
|
|
| 148 |
|
| 149 |
|
| 150 |
class BASNetModel(PreTrainedModel):
|
| 151 |
+
config_class = BASNetConfig
|
| 152 |
+
|
| 153 |
def __init__(self, config: BASNetConfig) -> None:
|
| 154 |
super().__init__(config)
|
| 155 |
|