2024年3月13日
ChatGLM3 源码解析（五）
摘要： PrefixEncoder # 根据前缀 ID 获取前缀嵌入 # 前缀嵌入将连接到分头之后的 K 和 V 上 class PrefixEncoder(torch.nn.Module): """ The torch.nn model to encode the prefix Input shape: 阅读全文
posted @ 2024-03-13 13:36 绝不原创的飞龙 阅读(154) 评论(0) 推荐(0) 编辑