From 6f8ce024d1ede51d03ece47c40f0e7b73e04488b Mon Sep 17 00:00:00 2001 From: Manh Nguyen <144032789+nguyen599@users.noreply.github.com> Date: Wed, 8 Oct 2025 22:27:01 +0700 Subject: [PATCH] Remove check_torch_compile_deepspeed (#3195) [skip ci] Signed-off-by: nguyen599 --- src/axolotl/utils/schemas/validation.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/src/axolotl/utils/schemas/validation.py b/src/axolotl/utils/schemas/validation.py index 0ec3e854f..4abe45e64 100644 --- a/src/axolotl/utils/schemas/validation.py +++ b/src/axolotl/utils/schemas/validation.py @@ -783,15 +783,6 @@ class OptimizationValidationMixin: return data - @model_validator(mode="before") - @classmethod - def check_torch_compile_deepspeed(cls, data): - if data.get("deepspeed") and data.get("torch_compile"): - raise ValueError( - "torch_compile should be set within your deepspeed config file" - ) - return data - @model_validator(mode="before") @classmethod def check_xentropy_patch_conflicts(cls, data):