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

be a bit pickier about loading dynamic prompt strategies

Browse Source
This commit is contained in:
Wing Lian
2024-12-30 16:52:41 -05:00
parent 2ed4246949
commit 03b86df506
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/axolotl/prompt_strategies/__init__.py
View File

@@ -20,9 +20,12 @@ def load(strategy, tokenizer, cfg, ds_cfg, processor=None):
if strategy.split(".")[-1].startswith("load_"): if strategy.split(".")[-1].startswith("load_"):
load_fn = strategy.split(".")[-1] load_fn = strategy.split(".")[-1]
strategy = ".".join(strategy.split(".")[:-1]) strategy = ".".join(strategy.split(".")[:-1])
else: elif len(strategy.split(".")) > 1:
try: try:
importlib.import_module(".".join(strategy.split(".")[:-1])) importlib.import_module(
"." + strategy.split(".")[-1],
".".join(strategy.split(".")[:-1]),
)
package = ".".join(strategy.split(".")[:-1]) package = ".".join(strategy.split(".")[:-1])
strategy = strategy.split(".")[-1] strategy = strategy.split(".")[-1]
except ModuleNotFoundError: except ModuleNotFoundError: