From 03b86df50655cd4e623067848a97432fab55a8bf Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 30 Dec 2024 16:52:41 -0500 Subject: [PATCH] be a bit pickier about loading dynamic prompt strategies --- src/axolotl/prompt_strategies/__init__.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/axolotl/prompt_strategies/__init__.py b/src/axolotl/prompt_strategies/__init__.py index ee8cc6145..645a9329c 100644 --- a/src/axolotl/prompt_strategies/__init__.py +++ b/src/axolotl/prompt_strategies/__init__.py @@ -20,9 +20,12 @@ def load(strategy, tokenizer, cfg, ds_cfg, processor=None): if strategy.split(".")[-1].startswith("load_"): load_fn = strategy.split(".")[-1] strategy = ".".join(strategy.split(".")[:-1]) - else: + elif len(strategy.split(".")) > 1: try: - importlib.import_module(".".join(strategy.split(".")[:-1])) + importlib.import_module( + "." + strategy.split(".")[-1], + ".".join(strategy.split(".")[:-1]), + ) package = ".".join(strategy.split(".")[:-1]) strategy = strategy.split(".")[-1] except ModuleNotFoundError: