Skip to content

Conversation

wqj2004
Copy link

@wqj2004 wqj2004 commented Feb 7, 2025

Overview

The whole RLHF training pipeline

onlinerlhf

Goal

In this PR, we need to finish the collector part in the above figure and add its unittest.

TODO

  • add vllm inferencer for LLM/VLM
  • scale up vllm inferencer in multi-GPUs
  • add several datasets definition @PaParaZz1 feature(nyz): add rlhf dataset #854
  • survey proper VQA dataset for RLHF training
  • add the final collector and test it on the above dataset
  • add tutorial and API doc

@PaParaZz1 PaParaZz1 changed the title feature(wqj): dev-vllm feature(wqj): add vllm collector Feb 8, 2025
@PaParaZz1 PaParaZz1 added the enhancement New feature or request label Feb 27, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants