feat: enable trl's autounwrap (#1060)

* feat: test trl's autounwrap

* fix: add check for adapter

* feat: add config to disable autounwrap

* chore: fix lint
This commit is contained in:
NanoCode012
2024-01-11 22:43:41 +09:00
committed by GitHub
parent 54fe07a905
commit b432889256
5 changed files with 24 additions and 10 deletions

2
.vscode/launch.json vendored
View File

@@ -11,7 +11,7 @@
"request": "launch",
"args": [
"-m", "axolotl.cli.train", "dev_sharegpt.yml",
// The flags below simplify debugging by overriding the axolotl config
// The flags below simplify debugging by overriding the axolotl config
// with the debugging tips above. Modify as needed.
"--dataset_processes=1", // limits data preprocessing to one process
"--max_steps=1", // limits training to just one step