Views
No views yet
git lfs install
git clone https://huggingface.co/lucky2me/Dorami1import torch
2from transformers import AutoTokenizer,AutoModelForCausalLM
3model_path = "The path of the model downloaded above"
4tokenizer = AutoTokenizer.from_pretrained(model_path)
5model = AutoModelForCausalLM.from_pretrained(model_path)
6text = "fill in any text you like."
7encoded_input = tokenizer(text, return_tensors='pt')
8output = model(**encoded_input)
9predicted_token_id = torch.argmax(output.logits[:, -1, :], dim=-1)
10decoded_text = tokenizer.decode(predicted_token_id, skip_special_tokens=True)
11print("decoded text:",decoded_text)