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