Seed Coder 8B Instruct

🚀 Seed-Coder-8B-Instruct

Seed-Coder是一個強大、透明且參數高效的8B規模開源代碼模型家族，包含基礎、指令和推理等多種變體。它通過以模型為中心的數據處理方式、透明的數據管道和出色的性能，推動了開源代碼模型的發展。本倉庫中的Seed-Coder-8B-Instruct模型具有特定的類型、訓練階段、數據源和上下文長度等特點。

🚀 快速開始

你需要安裝最新版本的transformers和accelerate：

pip install -U transformers accelerate

以下是一個簡單的示例，展示瞭如何使用Hugging Face的pipeline API加載模型並生成代碼：

from transformers import AutoTokenizer, AutoModelForCausalLM
import torch

model_id = "ByteDance-Seed/Seed-Coder-8B-Instruct"

tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True)

messages = [
    {"role": "user", "content": "Write a quick sort algorithm."},
]

input_ids = tokenizer.apply_chat_template(
    messages,
    tokenize=True,
    return_tensors="pt",
    add_generation_prompt=True,  
).to(model.device)

outputs = model.generate(input_ids, max_new_tokens=512)
response = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
print(response)

✨ 主要特性

• 以模型為中心：Seed-Coder主要利用大語言模型（LLMs）而非手工規則進行代碼數據過濾，最大限度地減少了預訓練數據構建中的人工工作量。
• 透明：我們公開分享了以模型為中心的數據管道的詳細信息，包括整理GitHub數據、提交數據和與代碼相關的網絡數據的方法。
• 強大：在各種編碼任務中，Seed-Coder在同類規模的開源模型中達到了最先進的性能。

📦 安裝指南

你需要安裝最新版本的transformers和accelerate：

pip install -U transformers accelerate

💻 使用示例

基礎用法

from transformers import AutoTokenizer, AutoModelForCausalLM
import torch

model_id = "ByteDance-Seed/Seed-Coder-8B-Instruct"

tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.bfloat16, device_map="auto", trust_remote_code=True)

messages = [
    {"role": "user", "content": "Write a quick sort algorithm."},
]

input_ids = tokenizer.apply_chat_template(
    messages,
    tokenize=True,
    return_tensors="pt",
    add_generation_prompt=True,  
).to(model.device)

outputs = model.generate(input_ids, max_new_tokens=512)
response = tokenizer.decode(outputs[0][input_ids.shape[-1]:], skip_special_tokens=True)
print(response)

📚 詳細文檔

模型下載

評估

Seed-Coder-8B-Instruct在廣泛的編碼任務中進行了評估，包括代碼生成、代碼推理、代碼編輯和軟件工程，在約8B的開源模型中達到了最先進的性能。

有關詳細的基準測試性能，請參考我們的📑 技術報告。

模型信息

📄 許可證

本項目採用MIT許可證。有關詳細信息，請參閱許可證文件。