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

prepare plugins needs to happen so registration can occur to build the plugin args (#2119)

Browse Source 
* prepare plugins needs to happen so registration can occur to build the plugin args

use yaml.dump

include dataset and more assertions

* attempt to manually register plugins rather than use fn

* fix fixture

* remove fixture

* move cli test to patched dir

* fix cce validation
This commit is contained in:
Wing Lian
2024-12-03 15:06:09 -05:00
committed by GitHub
parent 1ef70312ba
commit d87df2c776
4 changed files with 53 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/cutcrossentropy_install.py
View File

@@ -16,11 +16,11 @@ if v < V("2.4.0"):
sys.exit(0)
cce_spec = importlib.util.find_spec("cut_cross_entropy")
cce_spec_transformers = importlib.util.find_spec("cut_cross_entropy.transformers")
UNINSTALL_PREFIX = ""
if cce_spec and not cce_spec_transformers:
UNINSTALL_PREFIX = "pip uninstall -y cut-cross-entropy && "
if cce_spec:
if not importlib.util.find_spec("cut_cross_entropy.transformers"):
UNINSTALL_PREFIX = "pip uninstall -y cut-cross-entropy && "
print(
UNINSTALL_PREFIX