Views
No views yet


1from transformers import AutoModelForCausalLM, AutoTokenizer, TextStreamer, pipeline
2import torch
3
4model_id = "tarikkaankoc7/TKK-LLaMA3-8B-Elite-V1.0"
5
6model = AutoModelForCausalLM.from_pretrained(
7 model_id,
8 torch_dtype=torch.bfloat16,
9 device_map="auto",
10 trust_remote_code=True
11)
12
13tokenizer = AutoTokenizer.from_pretrained(
14 model_id,
15 trust_remote_code=True
16)
17
18streamer = TextStreamer(tokenizer)
19
20text_generation_pipeline = pipeline(
21 "text-generation",
22 model=model,
23 tokenizer=tokenizer,
24 model_kwargs={"torch_dtype": torch.bfloat16},
25 streamer=streamer
26)
27
28messages = [
29 {"role": "system", "content": "Sen yardımsever bir yapay zeka asistanısın ve kullanıcıların verdiği talimatlara doğrultusunda en iyi cevabı üretmeye çalışıyorsun."},
30 {"role": "user", "content": "Leonardo da Vinci'nin en ünlü tablosu hangisidir?"}
31]
32
33prompt = tokenizer.apply_chat_template(
34 messages,
35 tokenize=False,
36 add_generation_prompt=True
37)
38
39terminators = [
40 tokenizer.eos_token_id
41]
42
43outputs = text_generation_pipeline(
44 prompt,
45 max_new_tokens=2048,
46 eos_token_id=terminators,
47 do_sample=True,
48 temperature=0.6,
49 top_p=0.95
50)
51
52print(outputs[0]["generated_text"])Leonardo da Vinci'nin en ünlü tablosu Mona Lisa'dır.