Update modeling_CustomLEDForResultsId.py

modeling_CustomLEDForResultsId.py

@@ -5,7 +5,7 @@ from transformers.modeling_outputs import TokenClassifierOutput
 import torch.nn as nn
 
 class CustomLEDForResultsIdModel(LEDPreTrainedModel):
-    def __init__(self, config: LEDConfig, checkpoint): 
+    def __init__(self, config: LEDConfig, checkpoint=None): 
         super().__init__(config)
         self.num_labels = config.num_labels 
         print("Configs")