add missing __init__

This commit is contained in:
Wing Lian
2023-05-21 16:36:41 -04:00
parent 2809f3f21b
commit e0602a9e54

View File

@@ -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