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

vLLM doesn't estimate the model size properly #2788

Open
5 tasks done
Superskyyy opened this issue Feb 6, 2025 · 2 comments
Open
5 tasks done

vLLM doesn't estimate the model size properly #2788

Superskyyy opened this issue Feb 6, 2025 · 2 comments
Labels
⚡accelerate Related to accelerate 🐛 bug Something isn't working ⚡ PEFT Related to PEFT

Comments

@Superskyyy
Copy link
Contributor

Superskyyy commented Feb 6, 2025

Reproduction

In the current vLLM integration, the model size would be 0.000GB at startup (which I don't yet understand why), leading to an inaccurate profiling result and OOM after few steps. I will provide screenshot later.

System Info

  • Transformers version: 4.48.2
  • Accelerate version: 1.3.0
  • Accelerate config: not found
  • Datasets version: 3.2.0
  • HF Hub version: 0.27.1
  • TRL version: 0.15.0.dev0
  • bitsandbytes version: not installed
  • DeepSpeed version: 0.16.3
  • Diffusers version: not installed
  • Liger-Kernel version: 0.5.2
  • LLM-Blender version: not installed
  • OpenAI version: 1.60.2
  • PEFT version: 0.14.0
  • vLLM: 0.7.1

Checklist

  • I have checked that my issue isn't already filed (see open issues)
  • I have included my system information
  • Any code provided is minimal, complete, and reproducible (more on MREs)
  • Any code provided is properly formatted in code blocks, (no screenshot, more on code blocks)
  • Any traceback provided is complete
@github-actions github-actions bot added ⚡ PEFT Related to PEFT ⚡accelerate Related to accelerate 🐛 bug Something isn't working labels Feb 6, 2025
@Superskyyy
Copy link
Contributor Author

Superskyyy commented Feb 7, 2025

Screenshot:

Image

@xx-Jiangwen
Copy link

same question

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
⚡accelerate Related to accelerate 🐛 bug Something isn't working ⚡ PEFT Related to PEFT
Projects
None yet
Development

No branches or pull requests

2 participants