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

Bugfix: Fixes dtype issue described in #583 #584

Merged
merged 2 commits into from
Jan 25, 2024
Merged

Conversation

dnhkng
Copy link
Contributor

@dnhkng dnhkng commented Jan 25, 2024

Exllama generates logits in torch Half-dtype, but Outlines requires the Float-dtype.

This small change converts the logits to the required dtype (whatever that might be), solving issue #583.

Tested with Exllama on the example code on the github front page, and #583 is resolved.

@dnhkng dnhkng changed the title Bugfix: Fixes issue #583 Bugfix: Fixes dtype issue described in #583 Jan 25, 2024
@rlouf rlouf linked an issue Jan 25, 2024 that may be closed by this pull request
@rlouf rlouf added bug transformers Linked to the `transformers` integration labels Jan 25, 2024
@rlouf
Copy link
Member

rlouf commented Jan 25, 2024

Thank you so much for fixing the bug!

@rlouf rlouf merged commit 0cd9608 into outlines-dev:main Jan 25, 2024
5 checks passed
@rlouf rlouf linked an issue Feb 10, 2024 that may be closed by this pull request
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug transformers Linked to the `transformers` integration
Projects
None yet
Development

Successfully merging this pull request may close these issues.

All examples fail using ExllamaV2 Issue Running Dating Profile Example
2 participants