LinguaLaboratoriumMechanicus

Претрейн кастомной decoder-only модели на русскоязычном корпусе лора Warhammer 40,000.

Архитектура

Параметры ~163M
Слоёв 12
Размерность 768
Голов внимания 12
Контекст 1024
Vocab 50 257
FFN GELU, 4× expansion
Нормализация Pre-LayerNorm

Кастомная архитектура — загрузка с trust_remote_code=True.

Использование

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

repo_id = "GoldenGekko/LinguaLaboratoriumMechanicus"
device = "cuda" if torch.cuda.is_available() else "cpu"

tokenizer = AutoTokenizer.from_pretrained(repo_id)
model = AutoModelForCausalLM.from_pretrained(repo_id, trust_remote_code=True).to(device)

prompt = "В 31-м тысячелетии Империум"
inputs = tokenizer(prompt, return_tensors="pt").to(device)
output_ids = model.generate(
    inputs["input_ids"],
    max_new_tokens=100,
    do_sample=True,
    temperature=0.8,
    top_k=40,
    pad_token_id=tokenizer.pad_token_id,
    eos_token_id=tokenizer.eos_token_id,
)
print(tokenizer.decode(output_ids[0], skip_special_tokens=True))

Связанные репозитории

Citation

https://huggingface.co/GoldenGekko/LinguaLaboratoriumMechanicus
Downloads last month
32
Safetensors
Model size
0.2B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for GoldenGekko/LinguaLaboratoriumMechanicus

Finetunes
1 model