From e0602a9e549603c9ffce4d131c9103dc18b49034 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Sun, 21 May 2023 16:36:41 -0400 Subject: [PATCH] add missing __init__ --- src/axolotl/prompt_strategies/__init__.py | 11 +++++++++++ 1 file changed, 11 insertions(+) create mode 100644 src/axolotl/prompt_strategies/__init__.py 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