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

[Feature]: Qwen2.5 bitsandbytes support #8941

Open
1 task done
hanan9m opened this issue Sep 29, 2024 · 0 comments
Open
1 task done

[Feature]: Qwen2.5 bitsandbytes support #8941

hanan9m opened this issue Sep 29, 2024 · 0 comments

Comments

@hanan9m
Copy link

hanan9m commented Sep 29, 2024

🚀 The feature, motivation and pitch

Description:
Qwen2.5 (32B) is a state-of-the-art model, especially interesting in 4-bit precision (bitsandbytes).

  • I tried integrating it, but the model did not work as expected. the model output is just "!!!!!"
  • I created a Colab showing Qwen2.5 works in the transformers library but fails in vllm after my modification.
    in this notebook i show how the model is working using hugginface, and how after adding bitsandbytes support the output is gibberish
    i tried to add this lines, under Qwen2ForCausalLM class:
    bitsandbytes_stacked_params_mapping = {
        # shard_name, weight_name, index
        "q_proj": ("qkv_proj", 0),
        "k_proj": ("qkv_proj", 1),
        "v_proj": ("qkv_proj", 2),
        "gate_proj": ("gate_up_proj", 0),
        "up_proj": ("gate_up_proj", 1),
    }
  • There is similar PR just merge where adding bitsandbytes to Gemma2

bad output example

Prompt: 'The future of AI is', Generated text: '!!!!!!!!!!!!!!!!'

Alternatives

No response

Additional context

No response

Before submitting a new issue...

  • Make sure you already searched for relevant issues, and asked the chatbot living at the bottom right corner of the documentation page, which can answer lots of frequently asked questions.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

1 participant