Upload 4 files

config.json

@@ -13,7 +13,7 @@
   "torch_dtype": "float32",
   "global_step": 9975808,
   "auto_map": {
-    "AutoConfig": "configuration_mrs_paleta.MrPongConfig",
+    "AutoConfig": "configuration_mr_pong.MrPongConfig",
     "AutoModel": "modeling_mr_pong.MrPongForRL"
   }
 }

configuration_mr_pong.py

@@ -1,12 +1,12 @@
 """
-Configuration class for Mrs. Paleta Ping Pong RL Agent.
+Configuration class for MrPong Ping Pong RL Agent (fromziro/MrPong).
 """
 
 from transformers import PretrainedConfig
 
 
 class MrPongConfig(PretrainedConfig):
-    model_type = "mrs_paleta"
+    model_type = "mr_pong"
 
     def __init__(
         self,

modeling_mr_pong.py

@@ -1,5 +1,5 @@
 """
-PyTorch Modeling class for Mrs. Paleta Ping Pong RL Agent.
+PyTorch Modeling class for MrPong Ping Pong RL Agent (fromziro/MrPong).
 Compatible with Hugging Face AutoModel via trust_remote_code=True.
 """
 
@@ -14,13 +14,16 @@ from transformers.utils import ModelOutput
 try:
     from .configuration_mr_pong import MrPongConfig
 except ImportError:
-    from configuration_mr_pong import MrPongConfig
+    try:
+        from configuration_mr_pong import MrPongConfig
+    except ImportError:
+        from configuration_mrpong import MrPongConfig
 
 
 @dataclass
 class MrPongOutput(ModelOutput):
     """
-    Model output for Mr Pong.
+    Model output for MrPong.
     """
     logits: torch.FloatTensor = None
     value_estimate: Optional[torch.FloatTensor] = None