Fix decoder output class
#3
by
rom7
- opened
Fixes https://huggingface.co/kernels-community/mamba-ssm/discussions/2
This updates the decode method in utils/generation.py.
I believe at the rest of the places throughout the repo, a build would solve it? (but I am not sure)
This updates the decode method in utils/generation.py.
I believe at the rest of the places throughout the repo, a build would solve it? (but I am not sure)
rom7
changed pull request status to
open
Hey @rom7 , since we can't apply our own patches, I think it's better to open a PR upstream : https://github.com/state-spaces/mamba/blob/10b5d6358f27966f6a40e4bf0baa17a460688128/mamba_ssm/utils/generation.py#L14, once it's merged we can sync the new changes here.
Understood, thanks for the suggestion. I will close the PR here.
rom7
changed pull request status to
closed