先把diffusion拆掉试试
不然成本好高
反正本质也transformer
应该还ok吧
反正我这实验的主要目的也不是模型
只是模型输出要能不受通道限制