Commit Graph

34 Commits

Author SHA1 Message Date
bowei.fw 8208f986b9 . 2025-07-14 16:44:32 +08:00
bowei.fw d8bd161b1d . 2025-07-14 16:39:59 +08:00
博惟 724628eaf0 PullRequest: 355 [Lite] GRPO pre-commit 2: Refactor RemoteSGLangEngine thread and SGLang configuration
Merge branch fw/lite-fix1 of git@code.alipay.com:inclusionAI/AReaL.git into lite
https://code.alipay.com/inclusionAI/AReaL/pull_requests/355?tab=commit

Reviewed-by: 晓雷 <meizhiyu.mzy@antgroup.com>


* .
* .
* .
* .
* .
* .
* fix
* .
2025-07-14 15:20:17 +08:00
bowei.fw 037adedc70 . 2025-07-14 11:27:05 +08:00
博惟 d8038b2669 PullRequest: 354 [lite] GRPO pre-commit: minor changes in FSDP engine
Merge branch fw/lite-fix1 of git@code.alipay.com:inclusionAI/AReaL.git into lite
https://code.alipay.com/inclusionAI/AReaL/pull_requests/354

Reviewed-by: 晓雷 <meizhiyu.mzy@antgroup.com>


* .
* .
* .
* .
2025-07-14 11:19:40 +08:00
bowei.fw 932f9b9232 32k run 2025-07-13 21:48:48 +08:00
bowei.fw 34b6941a2f ready for boba 2025-07-12 19:20:02 +08:00
bowei.fw 0d03141cbc merge slurm launcher 2025-07-12 14:54:49 +08:00
meizhiyu.mzy 1af70c332c . 2025-07-12 12:27:06 +08:00
晓雷 e222cea659 . 2025-07-12 11:43:29 +08:00
meizhiyu.mzy 805437463f sglang server wrapper 2025-07-12 11:08:21 +08:00
bowei.fw 888751da38 qwen2 grpo works 2025-07-12 09:44:50 +08:00
meizhiyu.mzy 4a26f28adf . 2025-07-11 22:56:59 +08:00
晓雷 97511e43ff fix arg parse 2025-07-11 16:47:29 +08:00
晓雷 57ce1213ae merge ppo 2025-07-11 16:24:07 +08:00
晓雷 6acd1696f6 slurm launcher not tested 2025-07-11 16:14:59 +08:00
bowei.fw 0cbddb8aba . 2025-07-11 11:10:55 +08:00
Wei Fu 3bf9c85e40
[Fix] Merge previous contributions from fw/refactor to lite (#163)
* initial proposal

* add arealite

* .

* change api

* .

* remove LOG_ROOT

* remove MODEL_SAVE_PATH

* remove PARAM_REALLOC_PATH, DATASET_CACHE

* prepare for testing

* prepare for testing

* ready for run

* local run

* tests mainly pass

* format

* .

* amend cluster.py

* .

* .

* client test pass

* pass rollout test

* remove unused imports

* add arealite readme

* change api

* .

* .

* .

* .

* .

* .

* .

* .

* format

* .

* implement iteraptable generation (#112)

Co-authored-by: zhaochenyang <zhaochenyang20@gmail.com>

* .

* fix

* .

* .

* .

* pass controller generate batch test

* .

* refactor rollout controller into worker and controller

* .

* .

* .

* change to async rollout

* pass rollout controller test

* pass test

* .

* update readme

* .

* sft debug

* .

* add lisence

* remove unused files

* remove unsed args in ppo

* add hf engine wrapper  (#116)

* add hf engine

* fix issues

* fix ppo bugs and add test

* add hf client interface and modify cli args

* fix bugs

* fix issues

* Merge fw/refactor

* Finish hf wrapper test

* add test

---------

Co-authored-by: Wei Fu <36355462+garrett4wade@users.noreply.github.com>

* format

* format

* .

* refine hf engine

* .

* fix

* add fsdp engine and sft tests

* .

* .

* .

* pass ppo unittest

* pass ppo and rollout controller tests

* clear unused imports

* rename ppo to grpo

* change reward function organization

* reorganize code

* add dataset api

* .

* .

* .

* format

* chmod fix

* .

* rename workflow to collector

* refactor llm_client location

* .

* .

* fix llm server api

* refactor config structure

* .

* fix tests

* .

* .

* .

* Fix unresolved issue in SFTTrainer PR (#139)

* .

* .

* efficient loading

* format

* .

* .

* .

* .

* .

* .

* Add CI for testing AReaLite (#150)

* ci: add test-arealite

* ci: add checkout before running test-arealite

* ci: add USERNAME

* ci: add test script

* ci: add GitHub mirror

* ci: fix typo

* ci: clone one commit

* ci: fix condition

* ci: set command timeout to 60m

* ci: enable pip cache

* ci: optimize container lifecycle

* ci: split into many stages

* ci(test-arealite): fix typo

* ci: fix wrong env

* ci: fix pytest

* ci: uninstall transformer-engine

* ci: uninstall transformer-engine

* ci: fix model paths

* ci: show stdout/stderr

* ci: fix not clean up

* ci: backup sglang

* ci: remove tmp repo dir when run

* ci: fix docker run exit 1 condition

* ci(test-arealite): limit the concurrency and extend command timeout

* .

* merge fw/refactor

* revert some changes

* fix

---------

Co-authored-by: meizhiyu.mzy <meizhiyu.mzy@antgroup.com>
Co-authored-by: Chayenne <zhaochen20@outlook.com>
Co-authored-by: zhaochenyang <zhaochenyang20@gmail.com>
Co-authored-by: Jayon02 <qiujiangc@outlook.com>
Co-authored-by: root <meizhiyu.mzy>
Co-authored-by: Zijian Zhang <futrime@outlook.com>
2025-07-10 12:56:24 +08:00
博惟 c38cffc023 PullRequest: 340 [lite] Refactor trainer API into utilities and remove mb_spec in engine methods
Merge branch fw/lite-dev of git@code.alipay.com:inclusionAI/AReaL.git into lite
https://code.alipay.com/inclusionAI/AReaL/pull_requests/340

Reviewed-by: 晓雷 <meizhiyu.mzy@antgroup.com>


* support fsdp engine and sglang remote engine
* minor fix
* .
* refactor trainer
* add close
* rm mb_spec
* fix
2025-07-10 11:10:10 +08:00
bowei.fw 1bb23f2399 Merge branch 'fw/lite-dev' of code.alipay.com:inclusionAI/AReaL into fw/ppo 2025-07-10 11:08:47 +08:00
bowei.fw 520e38b1ee fix 2025-07-10 11:06:57 +08:00
bowei.fw 347bcc07a6 . 2025-07-10 11:05:08 +08:00
bowei.fw a78fd2dd24 rm mb_spec 2025-07-09 22:16:31 +08:00
bowei.fw b75aaaf130 refactor trainer 2025-07-09 21:48:28 +08:00
bowei.fw fe1f622841 . 2025-07-09 16:48:57 +08:00
bowei.fw 124184a7b7 minor fix 2025-07-09 16:48:11 +08:00
bowei.fw a6bcab22ba support fsdp engine and sglang remote engine 2025-07-09 16:45:29 +08:00
博惟 15dfbe837c PullRequest: 332 [lite] Support FSDP engines
Merge branch mzy/lite/engines of git@code.alipay.com:inclusionAI/AReaL.git into lite
https://code.alipay.com/inclusionAI/AReaL/pull_requests/332

Reviewed-by: 晓雷 <meizhiyu.mzy@antgroup.com>


* fsdp2 engine
* fix utils
* add fsdp engine test
* .
* fsdp engine test passed
* unsqueeze immediately before model inputs and after model outputts
* add optimizer save/load, add position id calculation for input
* .
* format
* not to squeeze
* add train and eval api
* .
* .
* improve fsdp engine data preprocessing
* format
* PullRequest: 337 [lite] Add SFT trainer example.
* trainer log
* minor changes
* add update weights from disk
* fix type annotation
2025-07-09 16:24:25 +08:00
博惟 8771778995 PullRequest: 331 [lite] Support remote sglang engine with corresponding testcases.
Merge branch fw/lite of git@code.alipay.com:inclusionAI/AReaL.git into lite
https://code.alipay.com/inclusionAI/AReaL/pull_requests/331

Reviewed-by: 晓雷 <meizhiyu.mzy@antgroup.com>


* .
* add test for sglang remote engine
* fix
2025-07-09 14:18:55 +08:00
garrett4wade cf0db6a43c format 2025-07-07 17:02:32 +08:00
garrett4wade 7ab6755379 Merge branch 'lite' of https://github.com/inclusionAI/AReaL into lite 2025-07-07 17:02:21 +08:00
博惟 e251abbbe8 add remote sglang engine 2025-07-07 17:01:35 +08:00
garrett4wade b6e19dbf60 format 2025-07-07 14:02:59 +08:00
博惟 1dfe91c470 add api 2025-07-07 13:47:41 +08:00