StarCoder2:大規模程式碼生成預訓練模型

Link:https://huggingface.co/bigcode/starcoder2-15b

StarCoder2是一個1500億引數的Transformer模型,在包括GitHub在內的600多種程式設計語言資料集上進行了預訓練,使用了Grouped Query Attention等技術。該模型可用於程式碼生成任務,支援多種程式設計語言。

需求人群:

  • 程式碼生成、程式碼補全

使用場景示例:

  • 使用StarCoder2基於提示生成Python程式碼
  • 使用StarCoder2進行程式碼搜尋任務的 Fine-tuning
  • 將StarCoder2部署到API Endpoint上提供程式碼生成服務

產品特色:

  • 支援600多種程式設計語言
  • 可用於程式碼生成任務
  • 支援CPU、GPU和混合精度訓練
  • 提供多種推理部署方案
返回頂端