Carrega o config dentro do arquivo

configuration_tjmg.py

@@ -1,10 +0,0 @@
-from transformers import PretrainedConfig
-
-# Configuração do Modelo (similar ao PretrainedConfig do Hugging Face)
-class CaptchaConfig(PretrainedConfig):
-    def __init__(self, input_dim=(40, 110), output_ndigits=5, output_vocab_size=10, vocab=None, **kwargs):
-        super().__init__(**kwargs)
-        self.input_dim = input_dim
-        self.output_ndigits = output_ndigits
-        self.output_vocab_size = output_vocab_size
-        self.vocab = vocab if vocab else [str(i) for i in range(10)]

modeling_tjmg.py

@@ -5,8 +5,21 @@ from transformers import PreTrainedModel
 from torchvision import transforms
 from PIL import Image
 
+from transformers import PretrainedConfig
+
+# Configuração do Modelo (similar ao PretrainedConfig do Hugging Face)
+class CaptchaConfig(PretrainedConfig):
+    def __init__(self, input_dim=(40, 110), output_ndigits=5, output_vocab_size=10, vocab=None, **kwargs):
+        super().__init__(**kwargs)
+        self.input_dim = input_dim
+        self.output_ndigits = output_ndigits
+        self.output_vocab_size = output_vocab_size
+        self.vocab = vocab if vocab else [str(i) for i in range(10)]
+
+
 # Modelo baseado na arquitetura do CaptchaCNN
 class CaptchaModel(PreTrainedModel):
+    config_class = CaptchaConfig
     model_type = "captcha"
 
     def __init__(self, config):