Add `device` and `dtype` parameters to `StableEmbedding` #87

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.

Sign up for GitHub

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

Jump to bottom

Merged

TimDettmers merged 1 commit into bitsandbytes-foundation:main from lostmsu:main

Jan 2, 2023

Contributor

lostmsu commented Nov 4, 2022

This allows using StableEmbedding anywhere where torch.nn.Embeding could be used


          add device and dtype parameters to StableEmbedding

62d39a2

Collaborator

TimDettmers commented Jan 2, 2023

Thank you! Also, the layer norm cast to default precision is a great addition!

TimDettmers merged commit 9d353ca into bitsandbytes-foundation:main

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet