TranscenderNing
TranscenderNing

reftTrainer重写了compute_loss函数,前向传播过程中输入除了input_id, attention_maks, labels, 还需要unit_locations,表示干预哪些位置token的隐藏状态
> 单测挂了  