Add Laguna Transformers config compatibility alias

lagunavision/backbones/hf_causal.py

@@ -17,6 +17,16 @@ from lagunavision.train.batching import pad_visual_batch
 
 def _install_huggingface_hub_dataclasses_shim() -> None:
     """Provide compatibility for Laguna remote code on HF Endpoint runtimes."""
+    try:
+        import transformers.configuration_utils as configuration_utils
+
+        if not hasattr(configuration_utils, "PreTrainedConfig") and hasattr(
+            configuration_utils, "PretrainedConfig"
+        ):
+            configuration_utils.PreTrainedConfig = configuration_utils.PretrainedConfig
+    except ImportError:
+        pass
+
     try:
         import huggingface_hub.dataclasses  # noqa: F401