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 |
|