Rename modeling.py to modeling_custom_mbz_test.py

modeling.py → modeling_custom_mbz_test.py

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