Views
No views yet
| Параметр | Значение |
|---|---|
| Объем параметров | ~150M |
| Архитектура | Llama |
| Размер словаря | 24,576 токенов |
| Длина контекста | 2048 токенов |
| Основной язык обучения | Русский |
| Перплексия | 22.59 (на валидационной выборке) |
| Лицензия | Apache 2.0 |
transformers. Пример кода для инференса на Python:1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4model_name = "Toondra/Clornet-150m"
5
6device = "cuda" if torch.cuda.is_available() else "cpu"
7tokenizer = AutoTokenizer.from_pretrained(model_name)
8model = AutoModelForCausalLM.from_pretrained(model_name).to(device)
9
10prompt = "В глубоком лесу росло странное дерево, которое"
11inputs = tokenizer(prompt, return_tensors="pt").to(device)
12
13with torch.no_grad():
14 outputs = model.generate(
15 **inputs,
16 max_new_tokens=64,
17 do_sample=True,
18 top_k=20,
19 top_p=0.8,
20 temperature=0.4,
21 repetition_penalty=1.15
22 )
23
24# Срезаем промпт, чтобы не дублировать его на выводе
25generated_ids = outputs[0][inputs.input_ids.shape[-1]:]
26generated_text = tokenizer.decode(generated_ids, skip_special_tokens=True)
27
28print("Результат генерации:")
29print(generated_text)