File size: 422 Bytes
e0602a9 ce34d64 e0602a9 0f74464 e0602a9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import importlib
def load(strategy, tokenizer, cfg):
try:
load_fn = "load"
if strategy.split(".")[-1].startswith("load_"):
load_fn = strategy.split(".")[-1]
strategy = ".".join(strategy.split(".")[:-1])
m = importlib.import_module(f".{strategy}", "axolotl.prompt_strategies")
fn = getattr(m, load_fn)
return fn(tokenizer, cfg)
except:
pass
|