diff --git a/src/axolotl/integrations/cut_cross_entropy/__init__.py b/src/axolotl/integrations/cut_cross_entropy/__init__.py index 7420674fa..916be61d0 100644 --- a/src/axolotl/integrations/cut_cross_entropy/__init__.py +++ b/src/axolotl/integrations/cut_cross_entropy/__init__.py @@ -72,7 +72,7 @@ class CutCrossEntropyPlugin(BasePlugin): if cfg.cut_cross_entropy: self._check_requirements() - from axolotl.integrations.cut_cross_entropy.monkeypatch.patch import ( + from .monkeypatch.patch import ( cce_patch, )