mirror of
https://github.com/lucasjinreal/Namo-R1.git
synced 2026-01-13 22:07:17 +08:00
9 lines
281 B
Python
9 lines
281 B
Python
from namo.r1.grpo import main
|
|
from trl import TrlParser
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = TrlParser((GRPOScriptArguments, GRPOConfig, ModelConfig))
|
|
script_args, training_args, model_args = parser.parse_args_and_config()
|
|
main(script_args, training_args, model_args)
|