Views
No views yet


1from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
2model_id = "Trendyol/Trendyol-LLM-7b-base-v1.0"
3tokenizer = AutoTokenizer.from_pretrained(model_id)
4model = AutoModelForCausalLM.from_pretrained(model_id,
5 device_map='auto',
6 load_in_8bit=True)
7sampling_params = dict(do_sample=True, temperature=0.3, top_k=50, top_p=0.9)
8pipe = pipeline("text-generation",
9 model=model,
10 tokenizer=tokenizer,
11 device_map="auto",
12 max_new_tokens=1024,
13 return_full_text=True,
14 repetition_penalty=1.1
15 )
16def generate_output(user_query):
17 outputs = pipe(user_query,
18 **sampling_params
19 )
20 return outputs[0]["generated_text"]
21user_query = "Ders çalışmanın en iyi 5 yolu:"
22response = generate_output(user_query)