winglian commited on
Commit
e0602a9
1 Parent(s): 2809f3f

add missing __init__

Browse files
src/axolotl/prompt_strategies/__init__.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import importlib
2
+ from functools import cache
3
+
4
+ @cache
5
+ def load(strategy, tokenizer, cfg):
6
+ try:
7
+ m = importlib.import_module(f".{strategy}", axolotl.prompt_strategies)
8
+ fn = getattr(m, "load")
9
+ return fn(tokenizer, cfg)
10
+ except:
11
+ pass