zxdu20's picture
Fix attention_mask and position_ids
373fd6b