HL 52437be1a6 [trainer] breaking: pass dataset as required args to SFTTrainer; also change ppo ray trainer to take custom datasets as inputs (#1282)

2025-05-02 21:03:22 -07:00

..

config

[config] feat: lr_warmup_steps (#564)

2025-03-14 16:09:12 +08:00

__init__.py

[dev] feat: immigrate from yapf & pylint to ruff based on pre-commit (#1010)

2025-04-18 07:49:31 -07:00

main_prime.py

feat: move AsyncLLM ChatCompletionScheduler to separate thread (#1274)

2025-04-27 22:02:52 +08:00

prime_core_algos.py

[Lint] fix: linting errors in all files (#1280)

2025-04-27 15:24:30 -07:00

prime_dp_rm.py

[Lint] fix: linting errors in all files (#1280)

2025-04-27 15:24:30 -07:00

prime_fsdp_workers.py

[Lint] fix: linting errors in all files (#1280)

2025-04-27 15:24:30 -07:00

prime_ray_trainer.py

[trainer] breaking: pass dataset as required args to SFTTrainer; also change ppo ray trainer to take custom datasets as inputs (#1282)

2025-05-02 21:03:22 -07:00

run_prime_qwen.sh

fix: remove deprecated remove_previous_ckpt key in prime_ray_trainer.py (#1254)

2025-04-25 18:12:18 +08:00