diff --git a/packages/prime-rl-configs/src/prime_rl/configs/trainer.py b/packages/prime-rl-configs/src/prime_rl/configs/trainer.py index a076d1e29c..4443fce229 100644 --- a/packages/prime-rl-configs/src/prime_rl/configs/trainer.py +++ b/packages/prime-rl-configs/src/prime_rl/configs/trainer.py @@ -208,7 +208,7 @@ class ModelConfig(BaseModelConfig): Field( description="Whether to compile the model using `torch.compile`.", ), - ] = None + ] = CompileConfig() ac: Annotated[ ActivationCheckpointConfig | None,