transformers
184f60dc - add attention_mask and position_ids in assisted model (#26892)

Commit
2 years ago
add attention_mask and position_ids in assisted model (#26892) * add attention_mask and position_ids in assisted model * fix bug * fix attention mask * fix attention_mask * check assist inputs * check assist input ids length * fix assist model type * set assist attention mask device
Author
Parents
Loading