Views
No views yet
1import warnings
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4path= f"sandeepsundaram/phi2-ultrachat-qlora"
5tokenizer = AutoTokenizer.from_pretrained(path)
6tokenizer.eos_token_id = model.config.eos_token_id
7tokenizer.pad_token = tokenizer.eos_token
8tokenizer.add_special_tokens({'pad_token': '[PAD]'})
9
10warnings.filterwarnings('ignore') # Ignore all warnings
11#inputs = tokenizer('Question: why human are cute then human? write in the form of poem. \n Output: ', return_tensors="pt", return_attention_mask=False).to('cuda')
12inputs = tokenizer('''write code for fibonaci series in python.''', return_tensors="pt", return_attention_mask=False).to('cuda')
13generation_params = {
14 'max_length': 512,
15 'do_sample': True,
16 'temperature': .5,
17 'top_p': 0.9,
18 'top_k': 50
19}
20
21outputs = model.generate(**inputs, **generation_params)
22decoded_outputs = tokenizer.batch_decode(outputs)
23
24for text in decoded_outputs:
25 text = text.replace('\\n', '\n')
26 print(text)
27 print("\n\n")