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

chore: improve error message

Browse Source
This commit is contained in:
NanoCode012
2025-06-19 18:22:04 +07:00
parent e97773d292
commit 533d5aaf4b
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/axolotl/integrations/cut_cross_entropy/__init__.py
View File

@@ -64,7 +64,9 @@ class CutCrossEntropyPlugin(BasePlugin):
"cut_cross_entropy.transformers" "cut_cross_entropy.transformers"
) )
if cce_spec_transformers is None: if cce_spec_transformers is None:
raise ImportError(_CCE_INSTALL_MESSAGE) raise ImportError(
"Transformers support is not installed. " + _CCE_INSTALL_MESSAGE
)
# Check if Axolotl's cce fork is installed # Check if Axolotl's cce fork is installed
try: try:
@@ -73,7 +75,10 @@ class CutCrossEntropyPlugin(BasePlugin):
if not AXOLOTL_CCE_FORK: if not AXOLOTL_CCE_FORK:
raise ImportError raise ImportError
except ImportError as e: except ImportError as e:
raise ImportError(_CCE_INSTALL_MESSAGE) from e raise ImportError(
"Axolotl's fork of cut_cross_entropy is not installed. "
+ _CCE_INSTALL_MESSAGE
) from e
def pre_model_load(self, cfg): def pre_model_load(self, cfg):
"""Apply cut cross entropy before model loading if enabled.""" """Apply cut cross entropy before model loading if enabled."""