Views
No views yet
phi-2 using SFT Training.
This model can answer information in a chat format as it is finetuned specifically on instructions specifically alpaca-cleanedMohamad Alhajarmicrosoft/phi-2### Instruction:
<prompt> (without the <>)
### Response:1from transformers import AutoTokenizer,AutoModelForCausalLM
2
3model_id = "malhajar/phi-2-chat"
4model = AutoModelForCausalLM.from_pretrained(model_name_or_path,
5 device_map="auto",
6 torch_dtype=torch.float16,
7 revision="main")
8
9tokenizer = AutoTokenizer.from_pretrained(model_id)
10
11question: "Türkiyenin en büyük şehir nedir?"
12# For generating a response
13prompt = '''
14### Instruction: {question} ### Response:
15'''
16input_ids = tokenizer(prompt, return_tensors="pt").input_ids
17output = model.generate(inputs=input_ids,max_new_tokens=512,pad_token_id=tokenizer.eos_token_id,top_k=50, do_sample=True,repetition_penalty=1.3
18 top_p=0.95,trust_remote_code=True,)
19response = tokenizer.decode(output[0])
20
21print(response)