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

[RWKV6] fix backward if h0 not passed #48

Merged

yzhangcs merged 1 commit into fla-org:main from hypnopump:patch-2

Aug 16, 2024

Contributor

hypnopump commented Aug 16, 2024

Currently, if initial_state is not passed as 0s, it breaks:

/fla/ops/rwkv6/chunk.py", line 753, in backward
    dh, dh0 = dh.to(q), dh0.to(q)
AttributeError: 'NoneType' object has no attribute 'to'

Fix it by only doing tensor ops if the initial_state is passed


          allow backward pass when initial state not passed

480fdc3

yzhangcs merged commit eb790cd into fla-org:main

1 check passed

Member

yzhangcs commented Aug 16, 2024

Thanks~

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

Labels

None yet