Czech GPT

Собственная decoder-only модель, обученная с нуля в учебном проекте. Экспорт этапа gpu_medium_depth12, шаг 11810; 46,152,704 параметров, 12 слоёв, контекст 256, собственный BPE 16k. Код и история обучения: https://github.com/Annprin/czech-gpt.

Это промежуточная модель: встречаются смысловые ошибки, повторы и смена темы. Связные рассказы и достоверные ответы пока не продемонстрированы.

В папке находятся веса float32, конфигурация и токенизатор. Состояния оптимизатора не нужны для инференса и в экспорт не входят. Исходный чекпоинт нужен для resume. Веса и ID токенизатора проверены после сохранения и повторной загрузки.

import torch
from transformers import AutoModelForCausalLM, AutoTokenizer

repo = "YOUR_USERNAME/czech-gpt"  # или путь к скачанной папке
tokenizer = AutoTokenizer.from_pretrained(repo)
model = AutoModelForCausalLM.from_pretrained(repo).eval()
inputs = tokenizer("Praha je", return_tensors="pt")
with torch.inference_mode():
    output = model.generate(**inputs, max_new_tokens=64, do_sample=False)
print(tokenizer.decode(output[0], skip_special_tokens=True))

Здесь from_pretrained загружает наши опубликованные веса для инференса; сторонние предобученные веса при обучении нашей модели не использовались.

Downloads last month
-
Safetensors
Model size
46.2M params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support