tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Remove check_torch_compile_deepspeed (#3195) [skip ci]

Browse Source 
Signed-off-by: nguyen599 <pnvmanh2123@gmail.com>
This commit is contained in:
Manh Nguyen
2025-10-08 22:27:01 +07:00
committed by GitHub
parent d0e9c3c1c5
commit 6f8ce024d1
1 changed files with 0 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/axolotl/utils/schemas/validation.py
View File

@@ -783,15 +783,6 @@ class OptimizationValidationMixin:
return data 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") @model_validator(mode="before")
@classmethod @classmethod
def check_xentropy_patch_conflicts(cls, data): def check_xentropy_patch_conflicts(cls, data):