Fix hidden_states dtype matching

#4
by li-plus - opened

Missed the .dtype before

Z.ai & THUKEG org

Thanks

Stanislas changed pull request status to merged

Sign up or log in to comment