Views
No views yet
1
2import torch
3from unsloth import FastLanguageModel
4from transformers import TextStreamer
5
6def run_inference():
7 # Model from finetune_tr_gsm8k_qwen14B.py
8 model_id = "uisikdag/qwen3-14b-instruct-gsm8k-tr-2"
9 print(f"Loading model from Hub: {model_id}")
10
11 # Load model and tokenizer from Hub
12 model, tokenizer = FastLanguageModel.from_pretrained(
13 model_name = model_id,
14 max_seq_length = 2048,
15 load_in_4bit = True,
16 )
17 FastLanguageModel.for_inference(model) # Enable native 2x faster inference
18
19 # 10 Turkish Math Questions (GSM8K style)
20 questions = [
21 "Bir manavda 56 elma vardı. Manav elmaların yarısını sattıktan sonra 12 tanesini daha sattı. Geriye kaç elma kaldı?",
22 "Ali'nin 5 kitabı var. Ayşe'nin kitap sayısı Ali'ninkinin 3 katından 2 eksik. Ayşe'nin kaç kitabı var?",
23 "Bir sınıfta 15 kız ve 10 erkek öğrenci var. Sınıftaki toplam öğrenci sayısının 2/5'i gözlüklü. Kaç öğrenci gözlüklü?",
24 "Bir çiftlikte tavuklar ve inekler var. Toplam baş sayısı 20, toplam ayak sayısı 56 ise kaç inek vardır?",
25 "3 kalem ve 2 silgi 50 TL. Bir kalem 10 TL ise bir silgi kaç TL'dir?",
26 "Bir araç saatte 60 km hızla 3 saat gitti. Sonra hızını 80 km'ye çıkarıp 2 saat daha gitti. Toplam kaç km yol aldı?",
27 "Ahmet bilyelerinin %20'sini Mehmet'e verince elinde 40 bilye kalıyor. Başlangıçta kaç bilyesi vardı?",
28 "Bir kutuda 5 kırmızı, 4 mavi ve 3 yeşil top var. Rastgele çekilen bir topun kırmızı olma olasılığı nedir?",
29 "Bir dikdörtgenin kısa kenarı 5 cm, uzun kenarı 12 cm'dir. Bu dikdörtgenin çevresi kaç cm'dir?",
30 "Babam 45 yaşında. Ben babamın yaşının 1/3'ü kadarım. 5 yıl sonra yaşlarımızın toplamı kaç olur?"
31 ]
32
33 print(f"Generating responses for {len(questions)} questions...\n")
34
35 for i, question in enumerate(questions, 1):
36 print(f"--- Question {i} ---")
37 print(f"User: {question}")
38
39 messages = [
40 {"role" : "user", "content" : question}
41 ]
42 text = tokenizer.apply_chat_template(
43 messages,
44 tokenize = False,
45 add_generation_prompt = True,
46 )
47
48 _ = model.generate(
49 **tokenizer(text, return_tensors = "pt").to("cuda"),
50 max_new_tokens = 512,
51 temperature = 0.7, top_p = 0.8, top_k = 20,
52 streamer = TextStreamer(tokenizer, skip_prompt = True),
53 )
54 print("\n" + "="*50 + "\n")
55
56
57if __name__ == "__main__":
58 run_inference()
591@misc{vonwerra2022trl,
2 title = {{TRL: Transformer Reinforcement Learning}},
3 author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
4 year = 2020,
5 journal = {GitHub repository},
6 publisher = {GitHub},
7 howpublished = {\url{https://github.com/huggingface/trl}}
8}