Fix hidden_states dtype matching

#4
by li-plus - opened

Missed the .dtype before

Knowledge Engineering Group (KEG) & Data Mining at Tsinghua University org

Thanks

Stanislas changed pull request status to merged

Sign up or log in to comment