facebookresearch/metaseq

Integrate LucidRain's RotaryEmbeddings

Open

#621 建立於 2023年1月27日

在 GitHub 查看
 (2 留言) (0 反應) (0 負責人)Python (701 fork)batch import
enhancementgood first issue

倉庫指標

Star
 (6,195 star)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

See https://github.com/lucidrains/rotary-embedding-torch/blob/main/rotary_embedding_torch/rotary_embedding_torch.py

And from PaLM paper:

We use RoPE embeddings (Su et al., 2021) rather than absolute or relative position embeddings, since RoPE embeddings have been shown to have better performance on long sequence lengths.

貢獻者指南

研究方向
閱讀連結的旋轉嵌入實作和 PaLM 論文。在程式碼庫中找到應用位置嵌入的地方（可能在注意力模組中），並將其替換為 RotaryEmbeddings。確保與現有模型架構相容。
技術棧
pythonpytorch
領域
machine learning
議題類型
功能
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-2 天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
大致清晰
前置要求
PythonPyTorchTransformer
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
50