openzeka_llm_beta / README.md
birolkuyumcu's picture
Update README.md
d330e13 verified
|
raw
history blame
1.12 kB
metadata
license: llama3

''' from transformers import AutoTokenizer, AutoModelForCausalLM import torch

model_id = "OPENZEKA/openzeka_llm_beta"

tokenizer = AutoTokenizer.from_pretrained(model_id) model = AutoModelForCausalLM.from_pretrained( model_id, torch_dtype=torch.bfloat16, device_map="auto", )

messages = [ {"role": "system", "content": "Sen Openzeka firmaının eğittiği asistansın sorulan sorulara doğru cevaplar ver "}, {"role": "user", "content": "Türkiye'nin yapay zeka alanında lider ülke olması için firmaların izleyeceği adımlar ne olmalıdır ?"}, ]

input_ids = tokenizer.apply_chat_template( messages, add_generation_prompt=True, return_tensors="pt" ).to(model.device)

terminators = [ tokenizer.eos_token_id, tokenizer.convert_tokens_to_ids("<|eot_id|>") ]

outputs = model.generate( input_ids, max_new_tokens=1024, eos_token_id=terminators, do_sample=True, repetition_penalty=1.1, temperature=0.6, top_p=0.9, ) response = outputs[0][input_ids.shape[-1]:] print(tokenizer.decode(response, skip_special_tokens=True)) '''