diff --git a/src/axolotl/prompt_strategies/__init__.py b/src/axolotl/prompt_strategies/__init__.py new file mode 100644 index 000000000..0bb936e4f --- /dev/null +++ b/src/axolotl/prompt_strategies/__init__.py @@ -0,0 +1,11 @@ +import importlib +from functools import cache + +@cache +def load(strategy, tokenizer, cfg): + try: + m = importlib.import_module(f".{strategy}", axolotl.prompt_strategies) + fn = getattr(m, "load") + return fn(tokenizer, cfg) + except: + pass