plugins:
- - axolotl.integrations.cut_cross_entropy.CutCrossEntropyPlugin
-
-cut_cross_entropy: truemain-base-py{python_version}-cu{cuda_version}-{pytorch_version}Tags examples:
main-base-py3.11-cu128-2.7.0main-base-py3.11-cu126-2.7.0main-base-py3.11-cu124-2.6.0main-base-py3.11-cu124-2.5.1main-base-py3.11-cu124-2.4.1# on push to main
main-py{python_version}-cu{cuda_version}-{pytorch_version}
-# latest main (currently torch 2.5.1, python 3.11, cuda 12.4)
+# latest main (currently torch 2.6.0, python 3.11, cuda 12.4)
main-latest
# nightly build
@@ -547,6 +549,7 @@ Tip
Tags examples:
main-py3.11-cu126-2.7.0main-py3.11-cu124-2.6.0main-py3.11-cu124-2.5.1main-py3.11-cu124-2.4.1