Gemago 2b

🚀 Gemago 2B模型卡片

Gemago 2B是一款基於Gemma的輕量級英韓互譯模型，能夠高效地實現英語與韓語之間的翻譯轉換。

🚀 快速開始

要快速運行該模型，首先確保你已經安裝了必要的庫。運行以下命令更新transformers庫：

pip install -U transformers

然後，根據你的具體需求，複製並使用以下代碼片段。

基礎用法

你可以通過以下代碼在transformers庫中運行該模型：

from transformers import AutoTokenizer, AutoModelForCausalLM

tokenizer = AutoTokenizer.from_pretrained("devworld/gemago-2b")
model = AutoModelForCausalLM.from_pretrained("devworld/gemago-2b")

def gen(text, max_length):
    input_ids = tokenizer(text, return_tensors="pt")
    outputs = model.generate(**input_ids, max_length=max_length)
    return tokenizer.decode(outputs[0])

def e2k(e):
    input_text = f"English:\n{e}\n\nKorean:\n"
    return gen(input_text, 1024)

def k2e(k):
    input_text = f"Korean:\n{k}\n\nEnglish:\n"
    return gen(input_text, 1024)

你可以直接在Colab中運行此代碼：

✨ 主要特性

• 翻譯功能：支持英語和韓語之間的雙向翻譯。
• 上下文長度：模型在8192個標記的上下文長度上進行訓練，與Gemma相當。

📚 詳細文檔

模型信息

• 原始Gemma模型頁面：Gemma
• GitHub上的模型頁面：Gemago
• 資源和技術文檔：
  • 博客（韓語）
  • 原始Google的Gemma - 2B
  • GitHub上的訓練代碼：Gemma - EasyLM（原作者Beomi）
• 使用條款：條款
• 作者：原作者Google，由DevWorld進行微調

推理參數

推理時可設置的參數如下：

數據集

模型訓練使用的數據集為：traintogpb/aihub-koen-translation-integrated-base-10m

語言支持

支持的語言包括韓語（ko）和英語（en）

標籤

模型標籤：gemma

小部件示例

• 韓語到英語（K2E）：

Korean:
나라의 말이 중국과 달라 문자와 서로 통하지 아니하다.

English:

• 英語到韓語（E2K）：

English:
Mr. and Mrs. Dursley were proud to say that they were perfectly normal.

Korean:

📄 許可證

模型使用的許可證為：apache - 2.0和gemma