PKU-YuanGroup/ProLLaMA

无条件生成

ZJL0111 opened this issue · 2 comments

ZJL0111 commented

您好在利用ProLLaMA进行蛋白质的无条件生成时，模型的输入是什么呢？ 还有在无条件生成的对比实验中，基于ESM2的是如何实现的呢？

Lyu6PosHao commented

您好。无条件生成，就是让ProLLaMA_Stage1做推理，输入为"Seq=<"，详情见this。

ESM2的生成，代码见this。大概就是输入为全部是[MASK]的token串，让ESM2依次预测出所有[MASK]对应的氨基酸。

  • 👍1
ZJL0111 commented

您好。无条件生成，就是让ProLLaMA_Stage1做推理，输入为"Seq=<"，详情见this。

ESM2的生成，代码见this。大概就是输入为全部是[MASK]的token串，让ESM2依次预测出所有[MASK]对应的氨基酸。

多谢