add missing __init__
This commit is contained in:
11
src/axolotl/prompt_strategies/__init__.py
Normal file
11
src/axolotl/prompt_strategies/__init__.py
Normal 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
|
||||||
Reference in New Issue
Block a user