Stanislas's picture li-plus's picture
Fix hidden_states dtype matching (#4)
3cb3f8f