Views
No views yet
n_layer: 4n_head: 8n_embd: 512n_positions (Maksymalny kontekst): 512EleutherAI/pythia-14m (BPE), posiadający predefiniowany, dodany chat_template pod format uczycia konwersacyjnego z tagami User: oraz Assistant:.Uwaga: Model celowo nie był uczony matematyki, ani wykonywania obliczeń. Nie potrafi rozwiązywać skomplikowanych zagadek logicznych, jego przeznaczeniem jest szybkie testowanie mechanizmów Supervised Fine-Tuning bezpośrednio w notatniku (np. Google Colab).
transformers?User: ... \nAssistant:.1from transformers import pipeline, AutoTokenizer, GPT2LMHeadModel
2
3repo_id = "onedevelopment/oneai-1.2-38m
4
5tokenizer = AutoTokenizer.from_pretrained(repo_id)
6model = GPT2LMHeadModel.from_pretrained(repo_id)
7
8pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
9
10# Pamiętaj u użyciu odpowiedniego formatowania promptu!
11prompt = "User: Hello!\nAssistant:"
12
13outputs = pipe(
14 prompt,
15 max_new_tokens=40,
16 do_sample=True,
17 temperature=0.7,
18 pad_token_id=tokenizer.eos_token_id,
19 eos_token_id=tokenizer.eos_token_id
20)
21
22print(outputs[0]['generated_text'])