Dummy Unknown

🚀 用於單元測試和持續集成的虛擬模型

本項目的虛擬模型主要用於單元測試和持續集成，為測試流程提供穩定、可復現的模型環境，確保代碼在不同環境下的兼容性和穩定性。

🚀 快速開始

安裝依賴

確保你已經安裝了transformers庫，可以使用以下命令進行安裝：

pip install transformers

💻 使用示例

基礎用法

import json
import os
from transformers import RobertaConfig, RobertaForMaskedLM, TFRobertaForMaskedLM

DIRNAME = "./dummy-unknown"

config = RobertaConfig(10, 20, 1, 1, 40)

model = RobertaForMaskedLM(config)
model.save_pretrained(DIRNAME)

tf_model = TFRobertaForMaskedLM.from_pretrained(DIRNAME, from_pt=True)
tf_model.save_pretrained(DIRNAME)

# Tokenizer:

vocab = [
    "l",
    "o",
    "w",
    "e",
    "r",
    "s",
    "t",
    "i",
    "d",
    "n",
    "\u0120",
    "\u0120l",
    "\u0120n",
    "\u0120lo",
    "\u0120low",
    "er",
    "\u0120lowest",
    "\u0120newer",
    "\u0120wider",
    "<unk>",
]
vocab_tokens = dict(zip(vocab, range(len(vocab))))
merges = ["#version: 0.2", "\u0120 l", "\u0120l o", "\u0120lo w", "e r", ""]

vocab_file = os.path.join(DIRNAME, "vocab.json")
merges_file = os.path.join(DIRNAME, "merges.txt")
with open(vocab_file, "w", encoding="utf-8") as fp:
    fp.write(json.dumps(vocab_tokens) + "\n")
with open(merges_file, "w", encoding="utf-8") as fp:
    fp.write("\n".join(merges))