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

Check whether device support bfloat16 #2653

Merged
merged 7 commits into from
Oct 25, 2024

Conversation

lvhan028
Copy link
Collaborator

No description provided.

@lvhan028 lvhan028 changed the title fallback to float16 if device doesn't support bfloat16 Check whether device support bfloat16 Oct 24, 2024
@lvhan028 lvhan028 merged commit d8f9e35 into InternLM:main Oct 25, 2024
5 checks passed
AllentDan pushed a commit to AllentDan/lmdeploy that referenced this pull request Nov 13, 2024
* fallback to float16 if torch.cuda.is_bf16_supported is False

* fix coverting qwen2-awq failed

* update

* update

* rollback load.py

* update
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants