tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

support custom module prompt strategy for rl

Browse Source
This commit is contained in:
Wing Lian
2025-02-02 23:47:00 -05:00
parent 704ddd6ff1
commit 79159b4871
1 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/axolotl/prompt_strategies/base.py
View File

@@ -14,6 +14,16 @@ def load(strategy, cfg, module_base=None, **kwargs):
strategy = strategy + ".default" strategy = strategy + ".default"
load_fn = strategy.split(".")[-1] load_fn = strategy.split(".")[-1]
strategy = ".".join(strategy.split(".")[:-1]) strategy = ".".join(strategy.split(".")[:-1])
if len(strategy.split(".")) > 1:
try:
importlib.import_module(
"." + strategy.split(".")[-1],
".".join(strategy.split(".")[:-1]),
)
module_base = ".".join(strategy.split(".")[:-1])
strategy = strategy.split(".")[-1]
except ModuleNotFoundError:
pass
mod = importlib.import_module(f".{strategy}", module_base) mod = importlib.import_module(f".{strategy}", module_base)
func = getattr(mod, load_fn) func = getattr(mod, load_fn)
return func(cfg, **kwargs) return func(cfg, **kwargs)