make cce default to true when using the plugin (#2562) [skip ci]

2025-04-25 17:14:26 -04:00
parent 2c2563bc34
commit 53dbf97d85
3 changed files with 1 additions and 4 deletions
--- a/src/axolotl/integrations/cut_cross_entropy/README.md
+++ b/src/axolotl/integrations/cut_cross_entropy/README.md
@@ -27,8 +27,6 @@ pip3 uninstall -y cut-cross-entropy && pip3 install "cut-cross-entropy[transform
 ```yaml
 plugins:
  - axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
-
-cut_cross_entropy: true
 ```

 ## Supported Models
--- a/src/axolotl/integrations/cut_cross_entropy/args.py
+++ b/src/axolotl/integrations/cut_cross_entropy/args.py
@@ -28,7 +28,7 @@ class CutCrossEntropyArgs(BaseModel):
    Input args for Cut Cross Entropy.
    """

-    cut_cross_entropy: Optional[bool] = None
+    cut_cross_entropy: Optional[bool] = True

    @model_validator(mode="before")
    @classmethod