Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Enable dataloaders for VLLM-s (llava and blip2) #646

Merged
merged 4 commits into from
Oct 10, 2024
Merged

Conversation

nikg4
Copy link
Collaborator

@nikg4 nikg4 commented Oct 10, 2024

-- offloading image data loading to workers helps a lot with GPU utilization for TRL_SFT (now close to 100% , previously very uneven ~50%)
-- and tune few other params

Towards OPE-467, OPE-551

Copy link

linear bot commented Oct 10, 2024

OPE-467

OPE-551

@nikg4 nikg4 merged commit d5067f5 into main Oct 10, 2024
1 check passed
@nikg4 nikg4 deleted the xrdaukar/llava-dl branch October 10, 2024 22:46
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants