Views
No views yet
transformers API or to load and use the model using the HuggingFace transformers library.transformers installed.pip uninstall -y transformers && pip install git+https://github.com/huggingface/transformers1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("susnato/phi-2")
4tokenizer = AutoTokenizer.from_pretrained("susnato/phi-2")
5
6inputs = tokenizer('''def print_prime(n):
7 """
8 Print all primes between 1 and n
9 """''', return_tensors="pt", return_attention_mask=False)
10
11outputs = model.generate(**inputs, max_length=200)
12text = tokenizer.batch_decode(outputs)[0]
13print(text)
14