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

Unable to Load Gemma2 Model with SGLANG #1869

Open
1 of 5 tasks
hahmad2008 opened this issue Nov 1, 2024 · 1 comment
Open
1 of 5 tasks

Unable to Load Gemma2 Model with SGLANG #1869

hahmad2008 opened this issue Nov 1, 2024 · 1 comment

Comments

@hahmad2008
Copy link

Checklist

  • 1. I have searched related issues but cannot get the expected help.
  • 2. The bug has not been fixed in the latest version.
  • 3. Please note that if the bug-related issue you submitted lacks corresponding environment info and a minimal reproducible demo, it will be challenging for us to reproduce and resolve the issue, reducing the likelihood of receiving feedback.
  • 4. If the issue you raised is not a bug but a question, please raise a discussion at https://github.com/sgl-project/sglang/discussions/new/choose Otherwise, it will be closed.
  • 5. Please use English, otherwise it will be closed.

Describe the bug

I am experiencing issues loading the Gemma2-9B model on a 24GB GPU using SGLANG. Despite trying various values for mem_fraction_static, I consistently encounter out-of-memory errors when increasing the value. If I decrease it, I receive a negative value for self.max_total_num_tokens.

In contrast, I can successfully load the same model on the 24GB GPU using VLLM with a GPU memory utilization of 0.95. Any guidance on resolving this would be appreciated.

Reproduction

Environment

sglang: 0.3.3.post1

@hahmad2008
Copy link
Author

@yileld sorry for tagging. Any idea?

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

No branches or pull requests

1 participant