File size: 268 Bytes
e0602a9
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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