Rename modeling_custom_mbz_test.py to modeling.py

modeling_custom_mbz_test.py → modeling.py

@@ -140,7 +140,7 @@ class GPT(nn.Module):
         x = self.out_head(self.norm(x))
         return x
 
-class CustomModelForCausalLM(PreTrainedModel):
+class CustomModel(PreTrainedModel):
     config_class = CustomConfig
     _supports_attention_backend = True