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

Update src/axolotl/prompt_strategies/base.py

Browse Source 
Co-authored-by: Wing Lian <wing.lian@gmail.com>
This commit is contained in:
Sung Ching Liu
2025-04-02 14:33:52 -04:00
parent ae8738aa87
commit 9b74298328
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -41,7 +41,10 @@ def load(strategy, cfg, module_base=None, **kwargs):
module_name = strategy.split(".")[-2] module_name = strategy.split(".")[-2]
mod = import_from_path(module_name, file_path) mod = import_from_path(module_name, file_path)
func = getattr(mod, load_fn) func = getattr(mod, load_fn)
return func(cfg, **kwargs) if func is None:
strategy = "." + ".".join(strategy.split(".")[:-1])
else:
return func(cfg, **kwargs)
except FileNotFoundError: except FileNotFoundError:
strategy = "." + ".".join(strategy.split(".")[:-1]) strategy = "." + ".".join(strategy.split(".")[:-1])