Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
2
3class TextGenerationAssistant:
4 def __init__(self, model_id:str):
5 self.tokenizer = AutoTokenizer.from_pretrained(model_id)
6 self.model = AutoModelForCausalLM.from_pretrained(model_id, device_map='auto',load_in_8bit=True,load_in_4bit=False)
7 self.pipe = pipeline("text-generation",
8 model=self.model,
9 tokenizer=self.tokenizer,
10 device_map="auto",
11 max_new_tokens=1024,
12 return_full_text=True,
13 repetition_penalty=1.0
14 )
15
16 self.sampling_params = dict(do_sample=True, temperature=0.5, top_k=50, top_p=0.9)
17 self.system_prompt = "Sen yardımcı bir asistansın. Sana verilen talimat ve girdilere en uygun cevapları üreteceksin. \n\n\n"
18
19 def format_prompt(self, user_input):
20 return "[INST] " + self.system_prompt + user_input + " [/INST]"
21
22 def generate_response(self, user_query):
23 prompt = self.format_prompt(user_query)
24 outputs = self.pipe(prompt, **self.sampling_params)
25 return outputs[0]["generated_text"].split("[/INST]")[1].strip()
26
27
28assistant = TextGenerationAssistant(model_id="Commencis/Commencis-LLM")
29
30# Enter your query here.
31user_query = "Faiz oranı yükseldiğinde kredi maliyetim nasıl etkilenir?"
32response = assistant.generate_response(user_query)
33print(response)
341from transformers import AutoTokenizer
2import transformers
3import torch
4
5model = "Commencis/Commencis-LLM"
6messages = [{"role": "user", "content": "Faiz oranı yükseldiğinde kredi maliyetim nasıl etkilenir?"}]
7
8tokenizer = AutoTokenizer.from_pretrained(model)
9prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
10pipeline = transformers.pipeline(
11 "text-generation",
12 model=model,
13 torch_dtype=torch.float16,
14 device_map="auto",
15)
16
17outputs = pipeline(prompt, max_new_tokens=1024, do_sample=True, temperature=0.5, top_k=50, top_p=0.9)
18print (outputs[0]["generated_text"].split("[/INST]")[1].strip())