Upload model.py with huggingface_hub

model.py

@@ -281,6 +281,8 @@ class BT4Config(PretrainedConfig):
         self.smol_activation = smol_activation
 
 class BT4(PreTrainedModel):
+    config_class = BT4Config
+    
     def __init__(self, config=None, embedding_size=1024, embedding_dense_sz=512, encoder_layers=15, encoder_d_model=1024, encoder_heads=32, encoder_dff=1536, dropout_rate=0.0, pol_embedding_size=1024, policy_d_model=1024, val_embedding_size=128, default_activation=Mish(),
                  use_smolgen=True, smol_hidden_channels=32, smol_hidden_sz=256, smol_gen_sz=256, smol_activation='swish'):
         # Initialize PreTrainedModel with config