Upload pretraining/pretrained_model.py with huggingface_hub

pretraining/pretrained_model.py

@@ -178,3 +178,10 @@ def lwm(*args, **kwargs) -> LWM:
     """Factory to preserve backward compatibility with older imports."""
 
     return LWM(*args, **kwargs)
+
+
+class PretrainedLWM(LWM):
+    """Alias retained for compatibility with existing inference scripts."""
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)