modifica config.json novamente

configuration_tjmg.py

@@ -0,0 +1,8 @@
+# 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)]

tjmg/config.json

@@ -2,6 +2,7 @@
   "model_type": "captcha",
   "architectures": ["CaptchaModel"],
   "auto_map": {
+    "AutoConfig": "configuration_tjmg.CaptchaConfig",
     "AutoModel": "modeling_tjmg.CaptchaModel"
   },
   "input_dim": [40, 110],