Put prompt_token_ids
, attentions_mask
and weights
on the same device
#719
+8
−6
prompt_token_ids
, attentions_mask
and weights
on the same device
#719