初始化项目,由ModelHub XC社区提供模型

Model: SupraLabs/Supra-50M-Base
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-06-25 21:59:04 +08:00
commit 3a121abd67
14 changed files with 161526 additions and 0 deletions

25
train_tokenizer.py Normal file
View File

@@ -0,0 +1,25 @@
print("[*] Loading libraries...")
from datasets import load_dataset
from tokenizers import ByteLevelBPETokenizer
from tqdm import tqdm
dataset = load_dataset("HuggingFaceFW/fineweb-edu", "sample-10BT", split="train", streaming=True)
def get_training_corpus():
dataset_iter = iter(dataset)
for _ in tqdm(range(500_000), desc="Feeding data"):
yield next(dataset_iter)["text"]
tokenizer = ByteLevelBPETokenizer()
print("[*] Training tokenizer...")
tokenizer.train_from_iterator(
get_training_corpus(),
vocab_size=32_000,
min_frequency=2,
show_progress=True,
special_tokens=["<s>", "<pad>", "</s>", "<unk>", "<mask>"]
)
tokenizer.save_model(".", "custom_llama_tokenizer")
print("[*] Tokenizer training complete!")